Searched defs:tensor_format (Results 1 – 5 of 5) sorted by relevance
437 TensorFormat tensor_format, char dimension) { in GetTensorDim()469 TensorFormat tensor_format, char dimension) { in GetTensorDim()485 inline int64_t GetTensorDim(const Tensor& tensor, TensorFormat tensor_format, in GetTensorDim()499 const std::vector<int64_t>& explicit_paddings, TensorFormat tensor_format, in GetExplicitPaddingForDim()
1220 string tensor_format; in FusedBatchNormOpBase() local1447 string tensor_format; in FusedBatchNormGradOpBase() local
674 string tensor_format; in MklFusedBatchNormOp() local1115 string tensor_format; in MklFusedBatchNormGradOp() local
458 Status CheckFormatConstraintsOnShape(const TensorFormat tensor_format, in CheckFormatConstraintsOnShape()
1517 cudnnTensorFormat_t tensor_format; in CheckAndFetchProjectionWeights() local1608 cudnnTensorFormat_t tensor_format; in Create() local