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 GetTensorDim(const Tensor& tensor, TensorFormat tensor_format, in GetTensorDim()499 const std::vector<int64>& explicit_paddings, TensorFormat tensor_format, in GetExplicitPaddingForDim()
1215 string tensor_format; in FusedBatchNormOpBase() local1437 string tensor_format; in FusedBatchNormGradOpBase() local
657 string tensor_format; in MklFusedBatchNormOp() local1061 string tensor_format; in MklFusedBatchNormGradOp() local
458 Status CheckFormatConstraintsOnShape(const TensorFormat tensor_format, in CheckFormatConstraintsOnShape()
1444 cudnnTensorFormat_t tensor_format; in CheckAndFetchProjectionWeights() local1535 cudnnTensorFormat_t tensor_format; in Create() local