| /external/tensorflow/tensorflow/core/kernels/ |
| bias_op_gpu.h | 33 int32 channel, TensorFormat data_format);
40 int32 channel, TensorFormat data_format);
|
| conv_grad_ops.h | 180 TensorFormat data_format);
189 Tensor* filter_backprop, TensorFormat data_format);
199 TensorFormat data_format);
208 Tensor* filter_backprop, TensorFormat data_format);
247 Padding padding, TensorFormat data_format,
256 Padding padding, TensorFormat data_format, ConvBackpropDimensions* dims);
|
| conv_ops.h | 40 TensorFormat data_format);
50 TensorFormat data_format);
|
| cudnn_pooling_gpu.h | 45 TensorFormat data_format, const Tensor& tensor_in,
60 TensorFormat data_format, const Tensor& out_backprop,
|
| pooling_ops_3d.h | 44 TensorFormat data_format,
75 TensorFormat data_format;
|
| pooling_ops_common_gpu.h | 46 TensorFormat data_format, const Tensor& tensor_in,
61 TensorFormat data_format, const Tensor* tensor_in,
|
| depthtospace_op.h | 43 template <typename Device, typename T, TensorFormat data_format>
|
| spacetodepth_op.h | 44 template <typename Device, typename T, TensorFormat data_format>
|
| pooling_ops_3d_gpu.h | 34 bool operator()(TensorFormat data_format, const T* bottom_data,
|
| xsmm_conv2d.h | 34 TensorFormat data_format);
|
| mkl_pooling_ops_common.cc | 31 TensorFormat data_format,
50 TensorFormat data_format,
65 TensorFormat data_format,
80 TensorFormat data_format) {
151 void ExtractMklOpParams(OpKernelContext* context, TensorFormat data_format,
|
| 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);
|
| mkl_pooling_ops_common.h | 62 TensorFormat data_format;
82 data_format(TensorFormat::FORMAT_NCHW) {}
87 TensorFormat data_format, const TensorShape& tensor_in_shape);
91 TensorFormat data_format, const MklShape* mkl_in_shape);
95 TensorFormat data_format, const MklDnnShape* mkl_in_shape);
102 TensorFormat data_format);
181 TensorFormat data_format_tf_;
416 void ExtractMklOpParams(OpKernelContext* context, TensorFormat data_format,
|
| maxpooling_op_gpu.h | 82 bool operator()(TensorFormat data_format, const T* bottom_data,
|
| bias_op.cc | 44 void GetBiasValueDims(const Tensor& value_tensor, TensorFormat data_format,
175 TensorFormat data_format_;
282 TensorFormat data_format_;
351 TensorFormat data_format_;
408 TensorFormat data_format_;
|
| pooling_ops_3d.cc | 56 Padding padding, TensorFormat data_format,
106 TensorFormat data_format, Padding padding_type,
121 TensorFormat data_format, Padding padding_type,
203 TensorFormat data_format_;
214 TensorFormat data_format, Tensor* output) {
377 TensorFormat data_format_;
389 TensorFormat data_format, Tensor* output) {
538 TensorFormat data_format_;
[all...] |
| conv_ops.cc | 64 Tensor* output, TensorFormat data_format) {
119 TensorFormat data_format) {
142 TensorFormat /*data_format*/) {
157 Tensor* output, TensorFormat data_format) {
198 TensorFormat data_format) {
212 Tensor* output, TensorFormat data_format) {
432 TensorFormat data_format_;
485 Tensor* output, TensorFormat data_format) {
[all...] |
| conv_grad_ops.cc | 101 Padding padding, TensorFormat data_format, ConvBackpropDimensions* dims) {
155 Padding padding, TensorFormat data_format,
|
| /external/tensorflow/tensorflow/core/util/ |
| tensor_format.h | 32 enum TensorFormat {
76 bool FormatFromString(const string& format_str, TensorFormat* format);
84 string ToString(TensorFormat format);
91 inline int GetTensorSpatialDims(int num_dims, TensorFormat format) {
110 TensorFormat format) {
130 inline int GetTensorBatchDimIndex(int num_dims, TensorFormat format) {
145 inline int GetTensorFeatureDimIndex(int num_dims, TensorFormat format) {
159 inline int GetTensorInnerFeatureDimIndex(int num_dims, TensorFormat format) {
165 inline int GetTensorSpatialDimIndex(int num_dims, TensorFormat format,
248 inline int32 GetTensorDimIndex(TensorFormat format, char dimension)
[all...] |
| tensor_format.cc | 36 string ToString(TensorFormat format) {
64 bool FormatFromString(const string& format_str, TensorFormat* format) {
|
| tensor_format_test.cc | 28 std::pair<TensorFormat, const char*> test_data_formats[] = {
105 GetTensorDimMap(const int num_spatial_dims, const TensorFormat format) {
137 TensorFormat format;
158 TensorFormat format = test_data_format.first;
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| fused_conv2d_bias_activation_op.h | 44 ActivationMode activation_mode, TensorFormat data_format,
58 ActivationMode activation_mode, TensorFormat data_format,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| bias_ops.cc | 69 TensorFormat data_format_;
113 TensorFormat data_format_;
|
| batch_norm_op.cc | 96 TensorFormat data_format_;
196 TensorFormat data_format_;
|
| /external/tensorflow/tensorflow/contrib/fused_conv/ops/ |
| fused_conv2d_bias_activation_op.cc | 58 TensorFormat data_format;
|
