| /external/tensorflow/tensorflow/core/kernels/ |
| fused_batch_norm_op.cc | 32 #include "tensorflow/core/util/tensor_format.h"
56 Tensor* saved_var_output, TensorFormat tensor_format,
58 OP_REQUIRES(context, tensor_format == FORMAT_NHWC,
139 TensorFormat tensor_format) {
140 OP_REQUIRES(context, tensor_format == FORMAT_NHWC,
226 Tensor* saved_inv_var, TensorFormat tensor_format,
231 const int64 batch_size = GetTensorDim(x, tensor_format, 'N');
232 const int64 channels = GetTensorDim(x, tensor_format, 'C');
233 const int64 height = GetTensorDim(x, tensor_format, 'H');
234 const int64 width = GetTensorDim(x, tensor_format, 'W')
509 string tensor_format; local
586 string tensor_format; local
[all...] |
| mkl_fused_batch_norm_op.cc | 22 #include "tensorflow/core/util/tensor_format.h"
54 string tensor_format; local
55 OP_REQUIRES_OK(context, context->GetAttr("data_format", &tensor_format));
56 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_),
241 const TensorFormat& tensor_format) {
249 : GetTensorDim(input, tensor_format, 'W'));
252 : GetTensorDim(input, tensor_format, 'H'));
255 : GetTensorDim(input, tensor_format, 'C'));
258 : GetTensorDim(input, tensor_format, 'N'));
260 GetStridesFromSizes(tensor_format, mkl_params.in_strides
351 string tensor_format; local
696 string tensor_format; local
1039 string tensor_format; local
[all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.cc | 1257 cudnnTensorFormat_t tensor_format; local
[all...] |
