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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
extract_image_patches_op.cc 70 int input_dim = GetTensorSpatialDimIndex(num_dims, data_format, i);
103 int input_dim = GetTensorSpatialDimIndex(num_dims, data_format, i);
135 const int64 dim = GetTensorSpatialDimIndex(num_dims, data_format, i);
conv_ops.cc 215 int input_dim = GetTensorSpatialDimIndex(num_dims(), data_format_, i);
272 const int64 dim = GetTensorSpatialDimIndex(num_dims(), data_format_, i);
366 int input_dim = GetTensorSpatialDimIndex(num_dims(), data_format_, i);
415 int64 dim = GetTensorSpatialDimIndex(num_dims(), data_format_, i);
521 int input_dim = GetTensorSpatialDimIndex(num_dims(), data_format_, i);
584 int64 dim = GetTensorSpatialDimIndex(num_dims(), data_format_, i);
pooling_ops.cc 204 int dim = GetTensorSpatialDimIndex(num_spatial_dims + 2, data_format, i);
443 int dim = GetTensorSpatialDimIndex(num_dims(), data_format_, i);
479 int dim = GetTensorSpatialDimIndex(num_dims(), data_format_, i);
  /external/tensorflow/tensorflow/core/util/
tensor_format.h 165 inline int GetTensorSpatialDimIndex(int num_dims, TensorFormat format,
417 dim_sizes[GetTensorSpatialDimIndex(dims, format, dim)] = spatial[dim];
tensor_format_test.cc 168 EXPECT_EQ(GetTensorSpatialDimIndex(num_dims, format, i), tdm.spatial(i));
  /external/tensorflow/tensorflow/core/kernels/
conv_grad_ops.cc 142 int image_dim = GetTensorSpatialDimIndex(num_dims, data_format, i);
  /external/tensorflow/tensorflow/core/framework/
common_shape_fns.cc 308 dims_actual[GetTensorSpatialDimIndex(num_dims, format, spatial_dim)] =
327 shape, GetTensorSpatialDimIndex(rank, format, spatial_dim_index));
352 out_dims[tensorflow::GetTensorSpatialDimIndex(
    [all...]

Completed in 167 milliseconds