OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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