| /external/tensorflow/tensorflow/compiler/tf2tensorrt/common/ |
| D | utils.cc | 148 const nvinfer1::TensorFormat& format) { in operator <<()
151 case nvinfer1::TensorFormat::kLINEAR: in operator <<()
155 case nvinfer1::TensorFormat::kCHW2: in operator <<()
159 case nvinfer1::TensorFormat::kHWC8: in operator <<()
163 case nvinfer1::TensorFormat::kCHW4: in operator <<()
167 case nvinfer1::TensorFormat::kCHW16: in operator <<()
171 case nvinfer1::TensorFormat::kCHW32: in operator <<()
176 case nvinfer1::TensorFormat::kDHWC8: in operator <<()
180 case nvinfer1::TensorFormat::kCDHW32: in operator <<()
184 case nvinfer1::TensorFormat::kHWC: in operator <<()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | algorithm_selector_test.cc | 48 nvinfer1::TensorFormat::kCHW32)); in TEST()
52 nvinfer1::TensorFormat::kCHW32)); in TEST()
54 nvinfer1::TensorFormat::kCHW32)); in TEST()
56 nvinfer1::TensorFormat::kCHW16)); in TEST()
72 nvinfer1::TensorFormat::kLINEAR)); in TEST()
76 nvinfer1::TensorFormat::kLINEAR)); in TEST()
78 nvinfer1::TensorFormat::kLINEAR)); in TEST()
80 nvinfer1::TensorFormat::kLINEAR)); in TEST()
82 nvinfer1::TensorFormat::kCHW16)); in TEST()
84 nvinfer1::TensorFormat::kCHW32)); in TEST()
|
| D | algorithm_selector.cc | 119 nvinfer1::TensorFormat input_format) const { in AllowShuffleAlgorithm()
123 return !(input_format == nvinfer1::TensorFormat::kLINEAR && in AllowShuffleAlgorithm()
130 return !(input_format == nvinfer1::TensorFormat::kCHW32 && in AllowShuffleAlgorithm()
|
| /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()
114 const TensorFormat& data_format, bool counts_include_padding) { 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 | bias_op.cc | 48 void GetBiasValueDims(const Tensor& value_tensor, TensorFormat data_format, in GetBiasValueDims()
145 TensorFormat data_format_;
224 TensorFormat data_format_;
282 TensorFormat data_format_;
335 BiasAddParams(const SpatialArray& in_shape, TensorFormat data_format, in BiasAddParams()
366 using ParamsDataType = std::tuple<SpatialArray, TensorFormat, DataType, int>;
376 TensorFormat data_format_;
532 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_t depth, int32_t channel, TensorFormat data_format);
43 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()
204 TensorFormat data_format_;
215 TensorFormat data_format, Tensor* output) { in launch()
387 TensorFormat data_format_;
399 TensorFormat data_format, Tensor* output) { in launch()
557 TensorFormat data_format_;
765 TensorFormat data_format_;
801 TensorFormat data_format, Padding padding_type, in launch()
[all …]
|
| D | conv_grad_shape_utils.cc | 100 TensorFormat data_format, ConvBackpropDimensions* dims) { in ConvBackpropComputeDimensionsV2()
165 Padding padding, TensorFormat data_format, in ConvBackpropComputeDimensions()
177 const TensorFormat& data_format, in Conv2DBackpropComputeInputShape()
|
| D | pooling_ops_common.cc | 61 TensorFormat format) { in operator ()()
77 TensorFormat format) { in operator ()()
122 TensorFormat data_format, in PoolParameters()
221 TensorFormat data_format, const Tensor& tensor_in, in Compute()
429 typename TTypes<T, 4, int>::Tensor out, TensorFormat data_format, \
444 TensorFormat data_format, const Tensor* tensor_in, const Tensor* tensor_out, in Compute()
475 TensorFormat transformed_input_data_format = data_format; in Compute()
|
| /external/tensorflow/tensorflow/core/util/autotune_maps/ |
| D | autotune_serialize_test.cc | 62 /*data_format=*/TensorFormat::FORMAT_NCHW, in TEST()
75 /*data_format=*/TensorFormat::FORMAT_NCHW, in TEST()
93 /*data_format=*/TensorFormat::FORMAT_NCHW, in TEST()
142 /*data_format=*/TensorFormat::FORMAT_NCHW, in TEST()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_utils.h | 97 tensorflow::TensorFormat data_format_tf,
106 ArrayAttr explicit_pad, tensorflow::TensorFormat data_format_tf,
111 tensorflow::Padding tf_pad, tensorflow::TensorFormat data_format_tf,
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_pooling_ops_common.h | 397 TensorFormat data_format;
423 data_format(TensorFormat::FORMAT_NCHW) {} in MklPoolParameters()
428 TensorFormat data_format, const TensorShape& tensor_in_shape);
431 TensorFormat data_format, const MklDnnShape* mkl_in_shape);
437 TensorFormat data_format);
566 if (pool_params->data_format == TensorFormat::FORMAT_NCHW) { in AllocateEmptyOutputTensor()
606 TensorFormat data_format_tf_;
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | pooling_ops.cc | 138 TensorFormat data_format_ = FORMAT_NHWC;
145 xla::TensorFormat XlaTensorFormat(tensorflow::TensorFormat data_format, in XlaTensorFormat()
155 return xla::TensorFormat(/*batch_dimension=*/batch_dimension, in XlaTensorFormat()
418 TensorFormat data_format_ = FORMAT_NHWC;
520 TensorFormat data_format_ = FORMAT_NHWC;
697 TensorFormat data_format_ = FORMAT_NHWC;
|