HomeSort by relevance Sort by last modified time
    Searched refs:dim_i (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
cudnn_pooling_gpu.cc 82 const auto dim_i = static_cast<se::dnn::DimIndex>(i); local
83 pooling_desc.set_window(dim_i, window[i]);
84 pooling_desc.set_stride(dim_i, stride[i]);
85 pooling_desc.set_padding(dim_i, padding[i]);
86 input_desc.set_spatial_dim(dim_i,
88 output_desc.set_spatial_dim(dim_i,
203 const auto dim_i = static_cast<se::dnn::DimIndex>(i); local
204 pooling_desc.set_window(dim_i, window[i]);
205 pooling_desc.set_stride(dim_i, stride[i]);
206 pooling_desc.set_padding(dim_i, padding[i])
    [all...]
  /external/tensorflow/tensorflow/core/util/
strided_slice_op.cc 260 int64 dim_i = input_shape.dim_size(i); local
265 if (dim_i == -1) {
273 {stride_i > 0 ? 0 : -1, stride_i > 0 ? dim_i : dim_i - 1}};
275 auto canonical = [stride_i, dim_i, masks, valid_range](int64 x, int c) {
279 int64 x_fwd = x < 0 ? dim_i + x : x; // make negative indices positive
299 int64 x_fwd = begin_i < 0 ? dim_i + begin_i : begin_i;
302 if (x_fwd < 0 || x_fwd >= dim_i) {
312 stride_i == 1 && begin_i == 0 && end_i == dim_i;
334 if (dim_i >= 0)
    [all...]
tensor_format_test.cc 60 int i() const { return dim_i; }
64 int dim_h, dim_w, dim_i, dim_o; member in struct:tensorflow::FilterDimMap
  /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
convert_nodes.cc     [all...]

Completed in 567 milliseconds