Home
last modified time | relevance | path

Searched refs:dim_i (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_pooling_gpu.cc82 const auto dim_i = static_cast<se::dnn::DimIndex>(i); in Compute() local
83 pooling_desc.set_window(dim_i, window[i]); in Compute()
84 pooling_desc.set_stride(dim_i, stride[i]); in Compute()
85 pooling_desc.set_padding(dim_i, padding[i]); in Compute()
86 input_desc.set_spatial_dim(dim_i, in Compute()
88 output_desc.set_spatial_dim(dim_i, in Compute()
218 const auto dim_i = static_cast<se::dnn::DimIndex>(i); in Compute() local
219 pooling_desc.set_window(dim_i, window[i]); in Compute()
220 pooling_desc.set_stride(dim_i, stride[i]); in Compute()
221 pooling_desc.set_padding(dim_i, padding[i]); in Compute()
[all …]
/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc277 int64 dim_i = input_shape.dim_size(i); in ValidateStridedSliceOp() local
282 if (dim_i == -1) { in ValidateStridedSliceOp()
290 {stride_i > 0 ? 0 : -1, stride_i > 0 ? dim_i : dim_i - 1}}; in ValidateStridedSliceOp()
292 auto canonical = [stride_i, dim_i, masks, valid_range](int64 x, int c) { in ValidateStridedSliceOp()
296 int64 x_fwd = x < 0 ? dim_i + x : x; // make negative indices positive in ValidateStridedSliceOp()
316 int64 x_fwd = begin_i < 0 ? dim_i + begin_i : begin_i; in ValidateStridedSliceOp()
319 if (x_fwd < 0 || x_fwd >= dim_i) { in ValidateStridedSliceOp()
329 stride_i == 1 && begin_i == 0 && end_i == dim_i; in ValidateStridedSliceOp()
351 if (dim_i >= 0) { in ValidateStridedSliceOp()
353 interval_length = -dim_i; in ValidateStridedSliceOp()
[all …]
Dtensor_format_test.cc60 int i() const { return dim_i; } in i()
64 int dim_h, dim_w, dim_i, dim_o; member
/external/llvm-project/polly/lib/External/isl/
Disl_schedule_tree.c1774 isl_size dim_i; in subtree_schedule_extend_from_children() local
1803 dim_i = range_dim(umap_i); in subtree_schedule_extend_from_children()
1804 if (dim_i < 0) { in subtree_schedule_extend_from_children()
1806 } else if (dim < dim_i) { in subtree_schedule_extend_from_children()
1807 umap = append_range(umap, dim_i - dim); in subtree_schedule_extend_from_children()
1808 dim = dim_i; in subtree_schedule_extend_from_children()
1809 } else if (dim_i < dim) { in subtree_schedule_extend_from_children()
1810 umap_i = append_range(umap_i, dim - dim_i); in subtree_schedule_extend_from_children()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc1584 int dim_i = input_type.getShape()[i]; in Verify() local
1585 if (begin_i > dim_i) { in Verify()
1587 "begin[{0}] cannot exceed dimension length: {1}", i, dim_i)); in Verify()
1589 if (size_i >= 0 && begin_i + size_i > dim_i) { in Verify()
1592 dim_i)); in Verify()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1774 int64_t dim_i = input_shape[i]; in CalculateSlicedShapeFromDenseIndices() local
1783 stride_i > 0 ? dim_i : dim_i - 1}; in CalculateSlicedShapeFromDenseIndices()
1791 point = point < 0 ? dim_i + point : point; in CalculateSlicedShapeFromDenseIndices()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc673 const nvinfer1::Dims dim_i = inputs.at(i).GetTrtDims(); in VerifyShapesMatch() local
674 if (dim_i.nbDims != dims_0.nbDims) { in VerifyShapesMatch()
680 if (dim_i.d[j] == -1 || dims_0.d[j] == -1) continue; in VerifyShapesMatch()
681 if (dim_i.d[j] != dims_0.d[j] && j != masked_dim) { in VerifyShapesMatch()