HomeSort by relevance Sort by last modified time
    Searched defs:dim_index (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
reduction_ops_common.cc 106 int dim_index = 0; local
107 for (; dim_index < data.dims(); ++dim_index) {
108 if (data.dim_size(dim_index) != 1) break;
110 if (dim_index >= data.dims()) {
114 // Starting from the (dim_index)-th dimension, dimensions
124 reduce_first_axis_ = bitmap[dim_index];
125 data_reshape_.push_back(data.dim_size(dim_index));
126 ++dim_index;
127 for (; dim_index < data.dims(); ++dim_index)
    [all...]
  /external/tensorflow/tensorflow/c/eager/
c_api_debug.cc 130 int64 dim_index = padded_shape.layout().minor_to_major(i); local
131 dev_dims.push_back(padded_shape.dimensions(dim_index));
159 TFE_TensorDebugInfo* debug_info, int dim_index) {
160 return debug_info->dev_dims[dim_index];
  /external/skia/modules/skottie/src/
SkottieAdapter.cpp 306 const auto dim_index = SkTPin<size_t>(static_cast<size_t>(fDimensions), local
313 fBlur->setSigma({ sigma * kDimensionsMap[dim_index].x(),
314 sigma * kDimensionsMap[dim_index].y() });
  /external/tensorflow/tensorflow/core/framework/
shape_inference.cc 716 int64 dim_index = dim_index_in; local
717 if (dim_index < 0) {
718 dim_index = s->dims_.size() + dim_index;
720 if (!FastBoundsCheck(dim_index, s->dims_.size())) {
722 return errors::InvalidArgument("Out of range dim_index ", dim_index_in,
727 dims[dim_index] = new_dim;
    [all...]
  /external/tensorflow/tensorflow/core/grappler/costs/
graph_properties.cc 915 int dim_index member in struct:tensorflow::grappler::SymbolicShapeRefiner::DimId
    [all...]

Completed in 188 milliseconds