Home
last modified time | relevance | path

Searched defs: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
203 const auto dim_i = static_cast<se::dnn::DimIndex>(i); in Compute() local
/external/tensorflow/tensorflow/core/util/
Dtensor_format_test.cc64 int dim_h, dim_w, dim_i, dim_o; member
Dstrided_slice_op.cc260 int64 dim_i = input_shape.dim_size(i); in ValidateStridedSliceOp() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc3545 auto dim_i = tensor_i->getDimensions(); in ConvertConcat() local