| /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 | 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 | 33 from tensorflow.python.debug.cli import tensor_format
309 return tensor_format.format_tensor(m, "m")
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 101 Tensor* saved_batch_var_output, TensorFormat tensor_format, in operator ()()
121 if (tensor_format == FORMAT_NCHW) { in operator ()()
122 const int64 in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()()
123 const int64 in_rows = GetTensorDim(x_input, tensor_format, 'H'); in operator ()()
124 const int64 in_cols = GetTensorDim(x_input, tensor_format, 'W'); in operator ()()
125 const int64 in_depths = GetTensorDim(x_input, tensor_format, 'C'); in operator ()()
217 if (tensor_format == FORMAT_NCHW) { in operator ()()
239 Tensor* saved_var_output, TensorFormat tensor_format, in operator ()()
259 if (tensor_format == FORMAT_NCHW) { in operator ()()
260 const int64 in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()()
[all …]
|
| D | fused_batch_norm_op.cu.cc | 127 template <typename T, typename U, TensorFormat tensor_format,
131 static_assert(tensor_format == FORMAT_NHWC || tensor_format == FORMAT_NCHW,
146 const int channel = (tensor_format == FORMAT_NHWC) in run()
176 template <TensorFormat tensor_format, bool add_side_input,
178 struct FusedBatchNormInferenceKernel<Eigen::half, float, tensor_format,
192 FusedBatchNormInferenceKernel<T, U, tensor_format, add_side_input,
222 if (tensor_format == FORMAT_NHWC) { in run()
262 const int32 channel = (tensor_format == FORMAT_NHWC) in run()
296 template <typename T, typename U, TensorFormat tensor_format,
303 FusedBatchNormInferenceKernel<T, U, tensor_format, add_side_input, in FusedBatchNormInferenceMetaKernel()
[all …]
|
| D | fused_batch_norm_op.h | 50 void operator()(OpKernelContext* context, TensorFormat tensor_format,
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 437 TensorFormat tensor_format, char dimension) { in GetTensorDim() argument
439 GetTensorDimIndex(tensor_format, dimension, dimension_attributes.size()); in GetTensorDim()
469 TensorFormat tensor_format, char dimension) { in GetTensorDim() argument
471 tensor_format, dimension); in GetTensorDim()
485 inline int64 GetTensorDim(const Tensor& tensor, TensorFormat tensor_format, in GetTensorDim() argument
487 return GetTensorDim(tensor.shape(), tensor_format, dimension); in GetTensorDim()
499 const std::vector<int64>& explicit_paddings, TensorFormat tensor_format, in GetExplicitPaddingForDim() argument
502 GetTensorDimIndex(tensor_format, dimension, explicit_paddings.size() / 2); in GetExplicitPaddingForDim()
|
| D | BUILD | 69 "tensor_format.cc",
70 "tensor_format.h",
179 "tensor_format.h",
356 "tensor_format.h",
539 name = "tensor_format",
540 srcs = ["tensor_format.cc"],
541 hdrs = ["tensor_format.h"],
556 ":tensor_format",
|
| /external/tensorflow/tensorflow/python/debug/ |
| D | BUILD | 289 name = "tensor_format",
290 srcs = ["cli/tensor_format.py"],
307 ":tensor_format",
392 ":tensor_format",
457 ":tensor_format",
948 ":tensor_format",
1168 ":tensor_format",
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 650 string tensor_format; in MklFusedBatchNormOp() local
651 OP_REQUIRES_OK(context, context->GetAttr("data_format", &tensor_format)); in MklFusedBatchNormOp()
652 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_), in MklFusedBatchNormOp()
1053 string tensor_format; in MklFusedBatchNormGradOp() local
1054 OP_REQUIRES_OK(context, context->GetAttr("data_format", &tensor_format)); in MklFusedBatchNormGradOp()
1055 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_), in MklFusedBatchNormGradOp()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 458 Status CheckFormatConstraintsOnShape(const TensorFormat tensor_format, in CheckFormatConstraintsOnShape() argument
462 if (tensor_format == FORMAT_NCHW_VECT_C) { in CheckFormatConstraintsOnShape()
466 shape_handle, GetTensorInnerFeatureDimIndex(num_dims, tensor_format)); in CheckFormatConstraintsOnShape()
|
| D | BUILD | 863 "//tensorflow/core/util:tensor_format",
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1339 cudnnTensorFormat_t tensor_format; in CheckAndFetchProjectionWeights() local
1344 /*dataType=*/&data_type, /*format=*/&tensor_format, in CheckAndFetchProjectionWeights()
1429 cudnnTensorFormat_t tensor_format; in Create() local
1434 /*dataType=*/&data_type, /*format=*/&tensor_format, in Create()
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 1661 "//tensorflow/core/util:tensor_format",
|