| /external/tensorflow/tensorflow/python/debug/cli/ |
| D | tensor_format_test.py | 28 from tensorflow.python.debug.cli import tensor_format
73 self.assertEqual({tensor_format.BEGIN_INDICES_KEY: subscripts},
105 tensor_format.locate_tensor_element(out, subscripts))
143 out = tensor_format.format_tensor(a, "a")
153 out = tensor_format.format_tensor(
161 out = tensor_format.format_tensor(
171 out = tensor_format.format_tensor(
186 out = tensor_format.format_tensor(a, "a")
198 out = tensor_format.format_tensor(a, None)
207 out = tensor_format.format_tensor(a, "a", include_metadata=True)
[all …]
|
| D | cli_shared.py | 27 from tensorflow.python.debug.cli import tensor_format
137 return tensor_format.HighlightOptions(
208 return tensor_format.format_tensor(
|
| D | stepper_cli.py | 28 from tensorflow.python.debug.cli import tensor_format
429 tensor_format.format_tensor(
593 tensor_lines = tensor_format.format_tensor(
|
| D | curses_ui.py | 35 from tensorflow.python.debug.cli import tensor_format
710 omitted, line_index, _, _ = tensor_format.locate_tensor_element(
|
| D | curses_ui_test.py | 31 from tensorflow.python.debug.cli import tensor_format
304 return tensor_format.format_tensor(m, "m")
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 73 Tensor* saved_var_output, TensorFormat tensor_format, in operator ()()
75 OP_REQUIRES(context, tensor_format == FORMAT_NHWC, in operator ()()
160 TensorFormat tensor_format) { in operator ()()
161 OP_REQUIRES(context, tensor_format == FORMAT_NHWC, in operator ()()
247 Tensor* saved_inv_var, TensorFormat tensor_format, in operator ()()
252 const int64 batch_size = GetTensorDim(x, tensor_format, 'N'); in operator ()()
253 const int64 channels = GetTensorDim(x, tensor_format, 'C'); in operator ()()
254 const int64 height = GetTensorDim(x, tensor_format, 'H'); in operator ()()
255 const int64 width = GetTensorDim(x, tensor_format, 'W'); in operator ()()
262 << " tensor format: " << tensor_format; in operator ()()
[all …]
|
| D | mkl_fused_batch_norm_op.cc | 503 string tensor_format; in MklFusedBatchNormOp() local
504 OP_REQUIRES_OK(context, context->GetAttr("data_format", &tensor_format)); in MklFusedBatchNormOp()
505 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_), in MklFusedBatchNormOp()
826 string tensor_format; in MklFusedBatchNormGradOp() local
827 OP_REQUIRES_OK(context, context->GetAttr("data_format", &tensor_format)); in MklFusedBatchNormGradOp()
828 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_), in MklFusedBatchNormGradOp()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 426 TensorFormat tensor_format, char dimension) { in GetTensorDim() argument
428 GetTensorDimIndex(tensor_format, dimension, dimension_attributes.size()); in GetTensorDim()
456 TensorFormat tensor_format, char dimension) { in GetTensorDim() argument
458 tensor_format, dimension); in GetTensorDim()
472 inline int64 GetTensorDim(const Tensor& tensor, TensorFormat tensor_format, in GetTensorDim() argument
474 return GetTensorDim(tensor.shape(), tensor_format, dimension); in GetTensorDim()
486 const std::vector<int64>& explicit_paddings, TensorFormat tensor_format, in GetExplicitPaddingForDim() argument
489 GetTensorDimIndex(tensor_format, dimension, explicit_paddings.size() / 2); in GetExplicitPaddingForDim()
|
| /external/tensorflow/tensorflow/python/debug/ |
| D | BUILD | 213 name = "tensor_format",
214 srcs = ["cli/tensor_format.py"],
231 ":tensor_format",
297 ":tensor_format",
331 ":tensor_format",
393 ":tensor_format",
735 ":tensor_format",
929 ":tensor_format",
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 312 Status CheckFormatConstraintsOnShape(const TensorFormat tensor_format, in CheckFormatConstraintsOnShape() argument
316 if (tensor_format == FORMAT_NCHW_VECT_C) { in CheckFormatConstraintsOnShape()
320 shape_handle, GetTensorInnerFeatureDimIndex(num_dims, tensor_format)); in CheckFormatConstraintsOnShape()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1229 cudnnTensorFormat_t tensor_format; in Create() local
1234 /*dataType=*/&data_type, /*format=*/&tensor_format, in Create()
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 964 "util/tensor_format.h",
|