Searched defs:tensor_format (Results 1 – 5 of 5) sorted by relevance
426 TensorFormat tensor_format, char dimension) { in GetTensorDim()456 TensorFormat tensor_format, char dimension) { in GetTensorDim()472 inline int64 GetTensorDim(const Tensor& tensor, TensorFormat tensor_format, in GetTensorDim()486 const std::vector<int64>& explicit_paddings, TensorFormat tensor_format, in GetExplicitPaddingForDim()
530 string tensor_format; in FusedBatchNormOp() local607 string tensor_format; in FusedBatchNormGradOp() local
503 string tensor_format; in MklFusedBatchNormOp() local826 string tensor_format; in MklFusedBatchNormGradOp() local
312 Status CheckFormatConstraintsOnShape(const TensorFormat tensor_format, in CheckFormatConstraintsOnShape()
1229 cudnnTensorFormat_t tensor_format; in Create() local