| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 37 enum TensorFormat { enum
101 bool FormatFromString(absl::string_view format_str, TensorFormat* format);
109 std::string ToString(TensorFormat format);
116 inline int GetTensorSpatialDims(int num_dims, TensorFormat format) { in GetTensorSpatialDims()
145 TensorFormat format) { in GetTensorDimsFromSpatialDims()
173 inline int GetTensorBatchDimIndex(int num_dims, TensorFormat format) { in GetTensorBatchDimIndex()
193 inline int GetTensorFeatureDimIndex(int num_dims, TensorFormat format) { in GetTensorFeatureDimIndex()
211 inline int GetTensorInnerFeatureDimIndex(int num_dims, TensorFormat format) { in GetTensorInnerFeatureDimIndex()
217 inline int GetTensorInnerWidthDimIndex(int num_dims, TensorFormat format) { in GetTensorInnerWidthDimIndex()
226 inline int GetTensorSpatialDimIndex(int num_dims, TensorFormat format, in GetTensorSpatialDimIndex()
[all …]
|
| D | tensor_format_test.cc | 28 std::pair<TensorFormat, const char*> test_data_formats[] = {
115 GetTensorDimMap(const int num_spatial_dims, const TensorFormat format) { in GetTensorDimMap()
152 TensorFormat format; in TEST()
173 TensorFormat format = test_data_format.first; in RunDimensionIndexesTest()
|
| D | tensor_format.cc | 40 string ToString(TensorFormat format) { in ToString()
76 bool FormatFromString(absl::string_view format_str, TensorFormat* format) { in FormatFromString()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling.h | 25 class TensorFormat {
27 TensorFormat(int batch_dimension, int feature_dimension, in TensorFormat() function
54 const TensorFormat& data_format);
60 const TensorFormat& data_format,
68 const TensorFormat& data_format);
75 const TensorFormat& data_format,
|
| D | pooling.cc | 32 const TensorFormat& data_format) { in AvgPoolDivideByCountWithGeneralPadding()
76 const TensorFormat& data_format) { in ComputeSums()
92 const TensorFormat& data_format) { in MakeSpatialPaddingConfig()
113 PrimitiveType dtype, const TensorFormat& data_format, in AvgPoolDivideByCount()
136 const TensorFormat& data_format) { in MaxPool()
151 const TensorFormat& data_format, in AvgPool()
176 const TensorFormat& data_format) { in MakeSpatialPadding()
197 const TensorFormat& data_format, in AvgPoolGrad()
|
| D | pooling_test.cc | 25 TensorFormat MakeNCHWFormat(int num_spatial_dims) { in MakeNCHWFormat()
30 return TensorFormat(/*batch_dimension=*/0, /*feature_dimension=*/1, in MakeNCHWFormat()
37 const xla::TensorFormat& data_format) { in MakeGeneralPadding()
50 const xla::TensorFormat& data_format) { in ExpandWithBatchAndFeatureDimensions()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_ops.h | 179 Tensor* in_backprop, TensorFormat data_format);
189 Tensor* filter_backprop, TensorFormat data_format);
200 TensorFormat data_format);
210 Tensor* filter_backprop, TensorFormat data_format);
|
| D | conv_grad_shape_utils.h | 74 Padding padding, TensorFormat data_format,
85 TensorFormat data_format, ConvBackpropDimensions* dims);
91 const TensorFormat& data_format,
|
| D | conv_ops.h | 41 TensorFormat data_format);
52 TensorFormat data_format);
76 TensorFormat data_format;
|
| D | depthwise_conv_op.h | 66 TensorFormat data_format);
73 TensorFormat data_format);
80 TensorFormat data_format);
88 TensorFormat data_format);
95 TensorFormat data_format);
102 TensorFormat data_format);
|
| D | cudnn_pooling_gpu.h | 45 TensorFormat data_format, const Tensor& tensor_in,
60 TensorFormat data_format, const Tensor& out_backprop,
|
| D | pooling_ops_3d.h | 44 TensorFormat data_format,
75 TensorFormat data_format;
|
| D | pooling_ops_common_gpu.h | 47 TensorFormat data_format, const Tensor& tensor_in,
63 TensorFormat data_format, const Tensor* tensor_in,
|
| D | bias_op_gpu.h | 35 int32 depth, int32 channel, TensorFormat data_format);
42 int32 depth, int32 channel, TensorFormat data_format);
|
| D | bias_op.cc | 45 void GetBiasValueDims(const Tensor& value_tensor, TensorFormat data_format, in GetBiasValueDims()
202 TensorFormat data_format_;
282 TensorFormat data_format_;
341 TensorFormat data_format_;
394 BiasAddParams(const SpatialArray& in_shape, TensorFormat data_format, in BiasAddParams()
425 using ParamsDataType = std::tuple<SpatialArray, TensorFormat, DataType, int>;
435 TensorFormat data_format_;
583 TensorFormat data_format_;
|
| D | pooling_ops_3d.cc | 51 Padding padding, TensorFormat data_format, in Pool3dParameters()
101 TensorFormat data_format, Padding padding_type, in launch()
116 TensorFormat data_format, Padding padding_type, in launch()
199 TensorFormat data_format_;
210 TensorFormat data_format, Tensor* output) { in launch()
373 TensorFormat data_format_;
385 TensorFormat data_format, Tensor* output) { in launch()
534 TensorFormat data_format_;
709 TensorFormat data_format_;
745 TensorFormat data_format, Padding padding_type, in launch()
[all …]
|
| D | conv_grad_shape_utils.cc | 99 TensorFormat data_format, ConvBackpropDimensions* dims) { in ConvBackpropComputeDimensionsV2()
160 Padding padding, TensorFormat data_format, in ConvBackpropComputeDimensions()
172 const TensorFormat& data_format, in Conv2DBackpropComputeInputShape()
|
| D | conv_ops_gpu.h | 104 TensorFormat data_format, int64 out_depths,
179 std::tuple<int64, int64, SpatialArray, TensorFormat, int64, SpatialArray,
215 TensorFormat data_format_;
|
| D | pooling_ops_common.cc | 61 TensorFormat format) { in operator ()()
77 TensorFormat format) { in operator ()()
121 TensorFormat data_format, in PoolParameters()
218 TensorFormat data_format, const Tensor& tensor_in, in Compute()
430 typename TTypes<T, 4, int>::Tensor out, TensorFormat data_format, \
445 TensorFormat data_format, const Tensor* tensor_in, const Tensor* tensor_out, in Compute()
466 TensorFormat transformed_input_data_format = data_format; in Compute()
|
| D | conv_ops_benchmark_test.cc | 78 TensorFormat data_format = FORMAT_NHWC) { in Conv2D()
109 int out_depth, TensorFormat data_format = FORMAT_NHWC) { in Conv2DWithBias()
136 TensorFormat data_format = FORMAT_NHWC) { in Conv2DWithBiasAndActivation()
158 int out_depth, TensorFormat data_format = FORMAT_NHWC) { in Conv2DWithBatchNorm()
196 TensorFormat data_format = FORMAT_NHWC) { in Conv2DWithBatchNormAndActivation()
221 TensorFormat data_format = FORMAT_NHWC) {
260 TensorFormat data_format = FORMAT_NHWC) {
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_utils.h | 79 tensorflow::Padding tf_pad, tensorflow::TensorFormat data_format_tf,
86 ArrayAttr explicit_pad, tensorflow::TensorFormat data_format_tf,
91 tensorflow::Padding tf_pad, tensorflow::TensorFormat data_format_tf,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | bias_ops.cc | 72 TensorFormat data_format_;
122 TensorFormat data_format_;
|
| D | pooling_ops.cc | 130 TensorFormat data_format_ = FORMAT_NHWC;
137 xla::TensorFormat XlaTensorFormat(tensorflow::TensorFormat data_format, in XlaTensorFormat()
147 return xla::TensorFormat(/*batch_dimension=*/batch_dimension, in XlaTensorFormat()
370 TensorFormat data_format_ = FORMAT_NHWC;
472 TensorFormat data_format_ = FORMAT_NHWC;
645 TensorFormat data_format_ = FORMAT_NHWC;
|
| D | depthtospace_op.cc | 43 TensorFormat data_format = data_format_; in Compile()
181 TensorFormat data_format_;
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_pooling_ops_common.h | 388 TensorFormat data_format;
414 data_format(TensorFormat::FORMAT_NCHW) {} in MklPoolParameters()
419 TensorFormat data_format, const TensorShape& tensor_in_shape);
422 TensorFormat data_format, const MklDnnShape* mkl_in_shape);
428 TensorFormat data_format);
551 if (pool_params->data_format == TensorFormat::FORMAT_NCHW) { in AllocateEmptyOutputTensor()
591 TensorFormat data_format_tf_;
|