Home
last modified time | relevance | path

Searched refs:dim_i (Results 1 – 4 of 4) 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()
203 const auto dim_i = static_cast<se::dnn::DimIndex>(i); in Compute() local
204 pooling_desc.set_window(dim_i, window[i]); in Compute()
205 pooling_desc.set_stride(dim_i, stride[i]); in Compute()
206 pooling_desc.set_padding(dim_i, padding[i]); in Compute()
[all …]
/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc260 int64 dim_i = input_shape.dim_size(i); in ValidateStridedSliceOp() local
265 if (dim_i == -1) { in ValidateStridedSliceOp()
273 {stride_i > 0 ? 0 : -1, stride_i > 0 ? dim_i : dim_i - 1}}; in ValidateStridedSliceOp()
275 auto canonical = [stride_i, dim_i, masks, valid_range](int64 x, int c) { in ValidateStridedSliceOp()
279 int64 x_fwd = x < 0 ? dim_i + x : x; // make negative indices positive in ValidateStridedSliceOp()
299 int64 x_fwd = begin_i < 0 ? dim_i + begin_i : begin_i; in ValidateStridedSliceOp()
302 if (x_fwd < 0 || x_fwd >= dim_i) { in ValidateStridedSliceOp()
312 stride_i == 1 && begin_i == 0 && end_i == dim_i; in ValidateStridedSliceOp()
334 if (dim_i >= 0) { in ValidateStridedSliceOp()
336 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/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc3545 auto dim_i = tensor_i->getDimensions(); in ConvertConcat() local
3546 if (dim_i.nbDims != dim.nbDims) { in ConvertConcat()
3553 if (j != index - 1 && dim_i.d[j] != dim.d[j]) { in ConvertConcat()