Searched refs:tensor_format_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 1196 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_), in FusedBatchNormOpBase()
1229 OP_REQUIRES(context, tensor_format_ == FORMAT_NHWC, in FusedBatchNormOpBase()
1271 const int64 in_batch = GetTensorDim(x, tensor_format_, 'N'); in ComputeWithReservedSpace()
1272 int64 in_planes = GetTensorDim(x, tensor_format_, '0'); in ComputeWithReservedSpace()
1273 int64 in_rows = GetTensorDim(x, tensor_format_, '1'); in ComputeWithReservedSpace()
1274 int64 in_cols = GetTensorDim(x, tensor_format_, '2'); in ComputeWithReservedSpace()
1275 const int64 in_depth = GetTensorDim(x, tensor_format_, 'C'); in ComputeWithReservedSpace()
1276 dest_shape = ShapeFromFormat(tensor_format_, in_batch, in ComputeWithReservedSpace()
1322 tensor_format_, use_reserved_space); in ComputeWithReservedSpace()
1328 tensor_format_, use_reserved_space); in ComputeWithReservedSpace()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 652 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_), in MklFusedBatchNormOp()
762 mkl_tensor_fmt = TFDataFormatToMklDnnDataFormat(tensor_format_); in Compute()
770 : TFShapeToMklDnnDimsInNCHW(src_tensor.shape(), tensor_format_); in Compute()
920 TensorFormat tensor_format_; member in tensorflow::MklFusedBatchNormOp
930 depth_ = static_cast<int>(GetTensorDim(input, tensor_format_, 'C')); in ExtractParams()
1055 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_), in MklFusedBatchNormGradOp()
1152 mkl_tensor_fmt = TFDataFormatToMklDnnDataFormat(tensor_format_); in Compute()
1164 : TFShapeToMklDnnDimsInNCHW(src_tensor.shape(), tensor_format_); in Compute()
1169 tensor_format_); in Compute()
1251 dnn_shape_diff_src.SetTfDimOrder(src_dims.size(), tensor_format_); in Compute()
[all …]
|