Home
last modified time | relevance | path

Searched refs:TensorFormat (Results 1 – 25 of 78) sorted by relevance

1234

/external/tensorflow/tensorflow/core/util/
Dtensor_format.h35 enum TensorFormat { enum
96 bool FormatFromString(const string& format_str, TensorFormat* format);
104 string ToString(TensorFormat format);
111 inline int GetTensorSpatialDims(int num_dims, TensorFormat format) { in GetTensorSpatialDims()
137 TensorFormat format) { in GetTensorDimsFromSpatialDims()
162 inline int GetTensorBatchDimIndex(int num_dims, TensorFormat format) { in GetTensorBatchDimIndex()
182 inline int GetTensorFeatureDimIndex(int num_dims, TensorFormat format) { in GetTensorFeatureDimIndex()
200 inline int GetTensorInnerFeatureDimIndex(int num_dims, TensorFormat format) { in GetTensorInnerFeatureDimIndex()
206 inline int GetTensorInnerWidthDimIndex(int num_dims, TensorFormat format) { in GetTensorInnerWidthDimIndex()
215 inline int GetTensorSpatialDimIndex(int num_dims, TensorFormat format, in GetTensorSpatialDimIndex()
[all …]
Dtensor_format_test.cc28 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()
Dtensor_format.cc40 string ToString(TensorFormat format) { in ToString()
74 bool FormatFromString(const string& format_str, TensorFormat* format) { in FormatFromString()
Dmkl_util.h224 void SetTfDimOrder(const size_t dimension, TensorFormat data_format) { in SetTfDimOrder()
267 TensorFormat data_format = FORMAT_NCHW; in IsTensorInNCHWFormat()
277 TensorFormat data_format = FORMAT_NHWC; in IsTensorInNHWCFormat()
416 TensorFormat MklDnn3DDataFormatToTFDataFormat(memory::format format);
417 TensorFormat MklDnnDataFormatToTFDataFormat(memory::format format);
677 inline void SetTfDimOrder(const size_t dimension, TensorFormat data_format) { in SetTfDimOrder()
701 TensorFormat data_format = MklDnnDataFormatToTFDataFormat(format); in SetTfDimOrder()
735 TensorFormat data_format = FORMAT_NCHW; in IsTensorInNCHWFormat()
745 TensorFormat data_format = FORMAT_NHWC; in IsTensorInNHWCFormat()
1071 inline void GetStridesFromSizes(TensorFormat data_format, size_t* strides, in GetStridesFromSizes()
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.h25 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,
Dpooling.cc32 const TensorFormat& data_format) { in AvgPoolDivideByCountWithGeneralPadding()
76 const TensorFormat& data_format) { in ComputeSums()
92 const TensorFormat& data_format) { in MakeSpatialPaddingConfig()
112 PrimitiveType dtype, const TensorFormat& data_format, in AvgPoolDivideByCount()
135 const TensorFormat& data_format) { in MaxPool()
150 const TensorFormat& data_format, in AvgPool()
175 const TensorFormat& data_format) { in MakeSpatialPadding()
196 const TensorFormat& data_format, in AvgPoolGrad()
Dpooling_test.cc25 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/
Dconv_grad_ops.h181 Tensor* in_backprop, TensorFormat data_format);
191 Tensor* filter_backprop, TensorFormat data_format);
202 TensorFormat data_format);
212 Tensor* filter_backprop, TensorFormat data_format);
263 Padding padding, TensorFormat data_format,
274 TensorFormat data_format, ConvBackpropDimensions* dims);
Ddepthwise_conv_op.h66 TensorFormat data_format);
73 TensorFormat data_format);
80 TensorFormat data_format);
88 TensorFormat data_format);
95 TensorFormat data_format);
102 TensorFormat data_format);
Dconv_ops.h41 TensorFormat data_format);
52 TensorFormat data_format);
76 TensorFormat data_format;
Dcudnn_pooling_gpu.h45 TensorFormat data_format, const Tensor& tensor_in,
60 TensorFormat data_format, const Tensor& out_backprop,
Dpooling_ops_3d.cc56 Padding padding, TensorFormat data_format, in Pool3dParameters()
106 TensorFormat data_format, Padding padding_type, in launch()
121 TensorFormat data_format, Padding padding_type, in launch()
203 TensorFormat data_format_;
214 TensorFormat data_format, Tensor* output) { in launch()
377 TensorFormat data_format_;
389 TensorFormat data_format, Tensor* output) { in launch()
538 TensorFormat data_format_;
713 TensorFormat data_format_;
749 TensorFormat data_format, Padding padding_type, in launch()
[all …]
Dpooling_ops_3d.h44 TensorFormat data_format,
75 TensorFormat data_format;
Dbias_op.cc46 void GetBiasValueDims(const Tensor& value_tensor, TensorFormat data_format, in GetBiasValueDims()
203 TensorFormat data_format_;
304 TensorFormat data_format_;
374 TensorFormat data_format_;
426 BiasAddParams(const SpatialArray& in_shape, TensorFormat data_format, in BiasAddParams()
457 using ParamsDataType = std::tuple<SpatialArray, TensorFormat, DataType, int>;
467 TensorFormat data_format_;
615 TensorFormat data_format_;
Dpooling_ops_common_gpu.h46 TensorFormat data_format, const Tensor& tensor_in,
61 TensorFormat data_format, const Tensor* tensor_in,
Dbias_op_gpu.h35 int32 depth, int32 channel, TensorFormat data_format);
42 int32 depth, int32 channel, TensorFormat data_format);
Dmkl_pooling_ops_common.h395 TensorFormat data_format;
421 data_format(TensorFormat::FORMAT_NCHW) {} in MklPoolParameters()
426 TensorFormat data_format, const TensorShape& tensor_in_shape);
430 TensorFormat data_format, const MklShape* mkl_in_shape);
434 TensorFormat data_format, const MklDnnShape* mkl_in_shape);
441 TensorFormat data_format);
561 if (pool_params->data_format == TensorFormat::FORMAT_NCHW) { in AllocateEmptyOutputTensor()
591 TensorFormat data_format_tf_;
771 void ExtractMklOpParams(OpKernelContext* context, TensorFormat data_format,
Dconv_ops_gpu.h88 TensorFormat data_format, int64 out_depths, in ConvParameters()
159 std::tuple<int64, int64, SpatialArray, TensorFormat, int64, SpatialArray,
195 TensorFormat data_format_;
Ddepthwise_conv_op_gpu.h603 TensorFormat data_format) { in LaunchDepthwiseConv2dGPUSmall()
681 TensorFormat data_format) {
701 TensorFormat data_format) {
719 TensorFormat data_format) {
742 TensorFormat data_format) {
778 TensorFormat data_format) {
802 TensorFormat data_format) {
966 TensorFormat data_format) {
997 TensorFormat data_format) {
1019 const T* filter, T* in_backprop, TensorFormat data_format) {
[all …]
Dmkl_pooling_ops_common.cc227 TensorFormat data_format, in Init()
255 TensorFormat data_format, in Init()
270 TensorFormat data_format, in Init()
295 TensorFormat data_format) { in Init()
405 void ExtractMklOpParams(OpKernelContext* context, TensorFormat data_format, in ExtractMklOpParams()
Davgpooling_op.cc102 TensorFormat data_format_;
179 TensorFormat data_format_;
357 TensorFormat data_format_;
430 TensorFormat data_format_;
561 TensorFormat data_format_;
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.h44 ActivationMode activation_mode, TensorFormat data_format,
58 ActivationMode activation_mode, TensorFormat data_format,
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc128 TensorFormat data_format_ = FORMAT_NHWC;
135 xla::TensorFormat XlaTensorFormat(tensorflow::TensorFormat data_format, in XlaTensorFormat()
145 return xla::TensorFormat(/*batch_dimension=*/batch_dimension, in XlaTensorFormat()
349 TensorFormat data_format_ = FORMAT_NHWC;
451 TensorFormat data_format_ = FORMAT_NHWC;
622 TensorFormat data_format_ = FORMAT_NHWC;
Dbias_ops.cc72 TensorFormat data_format_;
122 TensorFormat data_format_;
Ddepthtospace_op.cc43 TensorFormat data_format = data_format_; in Compile()
181 TensorFormat data_format_;

1234