Searched refs:dim_i (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_pooling_gpu.cc | 83 const auto dim_i = static_cast<se::dnn::DimIndex>(i); in Compute() local
84 pooling_desc.set_window(dim_i, window[i]); in Compute()
85 pooling_desc.set_stride(dim_i, stride[i]); in Compute()
86 pooling_desc.set_padding(dim_i, padding[i]); in Compute()
87 input_desc.set_spatial_dim(dim_i, in Compute()
89 output_desc.set_spatial_dim(dim_i, in Compute()
213 const auto dim_i = static_cast<se::dnn::DimIndex>(i); in Compute() local
214 pooling_desc.set_window(dim_i, window[i]); in Compute()
215 pooling_desc.set_stride(dim_i, stride[i]); in Compute()
216 pooling_desc.set_padding(dim_i, padding[i]); in Compute()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/shim/ |
| D | tensor_view.h | 109 inline int Dim(int dim_i) const { in Dim() argument
110 DCHECK(RANK > 0 && dim_i < RANK) << "dim: " << dim_i << " rank:" << RANK; in Dim()
112 if (dim_i < 0) dim_i = ((dim_i % RANK) + RANK) % RANK; in Dim()
113 return shape_[dim_i]; in Dim()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 281 int64_t dim_i = input_shape.dim_size(i); in ValidateStridedSliceOp() local
286 if (dim_i == -1) { in ValidateStridedSliceOp()
294 {stride_i > 0 ? 0 : -1, stride_i > 0 ? dim_i : dim_i - 1}}; in ValidateStridedSliceOp()
296 auto canonical = [stride_i, dim_i, masks, valid_range](int64_t x, int c) { in ValidateStridedSliceOp()
301 x < 0 ? dim_i + x : x; // make negative indices positive in ValidateStridedSliceOp()
321 int64_t x_fwd = begin_i < 0 ? dim_i + begin_i : begin_i; in ValidateStridedSliceOp()
324 if (x_fwd < 0 || x_fwd >= dim_i) { in ValidateStridedSliceOp()
334 stride_i == 1 && begin_i == 0 && end_i == dim_i; in ValidateStridedSliceOp()
356 if (dim_i >= 0) { in ValidateStridedSliceOp()
358 interval_length = -dim_i; in ValidateStridedSliceOp()
[all …]
|
| D | tensor_format_test.cc | 60 int i() const { return dim_i; } in i()
64 int dim_h, dim_w, dim_i, dim_o; member
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 2054 int dim_i = input_type.getShape()[i]; in verify() local
2055 if (begin_i > dim_i) { in verify()
2057 "begin[{0}] cannot exceed dimension length: {1}", i, dim_i)); in verify()
2059 if (size_i >= 0 && begin_i + size_i > dim_i) { in verify()
2062 dim_i)); in verify()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 1914 int64_t dim_i = input_shape[i]; in CalculateSlicedShapeFromDenseIndices() local
1923 stride_i > 0 ? dim_i : dim_i - 1}; in CalculateSlicedShapeFromDenseIndices()
1931 point = point < 0 ? dim_i + point : point; in CalculateSlicedShapeFromDenseIndices()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5577 Value dim_i = rewriter.create<arith::IndexCastOp>( in matchAndRewrite() local
5580 end_indices.push_back(dim_i); in matchAndRewrite()
5582 shape_values.push_back(dim_i); in matchAndRewrite()
5585 Value dim_i = rewriter.create<arith::ConstantOp>( in matchAndRewrite() local
5588 end_indices.push_back(dim_i); in matchAndRewrite()
5590 shape_values.push_back(dim_i); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 640 const nvinfer1::Dims dim_i = inputs.at(i).GetTrtDims(); in VerifyShapesMatch() local
641 if (dim_i.nbDims != dims_0.nbDims) { in VerifyShapesMatch()
647 if (dim_i.d[j] == -1 || dims_0.d[j] == -1) continue; in VerifyShapesMatch()
648 if (dim_i.d[j] != dims_0.d[j] && j != masked_dim) { in VerifyShapesMatch()
|