| /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 | BUILD | 52 ":tensor_format",
85 name = "tensor_format",
86 srcs = ["tensor_format.py"],
103 ":tensor_format",
196 ":tensor_format",
274 ":tensor_format",
|
| 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 ()()
130 if (tensor_format == FORMAT_NCHW) { in operator ()()
131 const int64_t in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()()
132 const int64_t in_rows = GetTensorDim(x_input, tensor_format, 'H'); in operator ()()
133 const int64_t in_cols = GetTensorDim(x_input, tensor_format, 'W'); in operator ()()
134 const int64_t in_depths = GetTensorDim(x_input, tensor_format, 'C'); in operator ()()
226 if (tensor_format == FORMAT_NCHW) { in operator ()()
248 Tensor* saved_var_output, TensorFormat tensor_format, in operator ()()
277 if (tensor_format == FORMAT_NCHW) { in operator ()()
278 const int64_t in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()()
[all …]
|
| D | fused_batch_norm_op.cu.cc | 125 template <typename T, typename U, TensorFormat tensor_format,
129 static_assert(tensor_format == FORMAT_NHWC || tensor_format == FORMAT_NCHW,
144 const int channel = (tensor_format == FORMAT_NHWC) in run()
174 template <TensorFormat tensor_format, bool add_side_input,
176 struct FusedBatchNormInferenceKernel<Eigen::half, float, tensor_format,
190 FusedBatchNormInferenceKernel<T, U, tensor_format, add_side_input,
220 if (tensor_format == FORMAT_NHWC) { in run()
260 const int32 channel = (tensor_format == FORMAT_NHWC) in run()
294 template <typename T, typename U, TensorFormat tensor_format,
301 FusedBatchNormInferenceKernel<T, U, tensor_format, add_side_input, in FusedBatchNormInferenceMetaKernel()
[all …]
|
| D | fused_batch_norm_op.h | 44 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 | 73 "tensor_format.cc",
74 "tensor_format.h",
184 "tensor_format.h",
361 "tensor_format.h",
545 name = "tensor_format",
546 srcs = ["tensor_format.cc"],
547 hdrs = ["tensor_format.h"],
562 ":tensor_format",
|
| /external/tensorflow/tensorflow/core/util/autotune_maps/ |
| D | conv_parameters.proto | 54 // third_party/tensorflow/core/util/tensor_format.h.
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 657 string tensor_format; in MklFusedBatchNormOp() local
658 OP_REQUIRES_OK(context, context->GetAttr("data_format", &tensor_format)); in MklFusedBatchNormOp()
659 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_), in MklFusedBatchNormOp()
1061 string tensor_format; in MklFusedBatchNormGradOp() local
1062 OP_REQUIRES_OK(context, context->GetAttr("data_format", &tensor_format)); in MklFusedBatchNormGradOp()
1063 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_), in MklFusedBatchNormGradOp()
|
| /external/tensorflow/tensorflow/python/debug/wrappers/ |
| D | BUILD | 60 "//tensorflow/python/debug/cli:tensor_format",
|
| /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 | 902 "//tensorflow/core/util:tensor_format",
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1444 cudnnTensorFormat_t tensor_format; in CheckAndFetchProjectionWeights() local
1449 /*dataType=*/&data_type, /*format=*/&tensor_format, in CheckAndFetchProjectionWeights()
1535 cudnnTensorFormat_t tensor_format; in Create() local
1540 /*dataType=*/&data_type, /*format=*/&tensor_format, in Create()
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 1692 "//tensorflow/core/util:tensor_format",
|