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()
1194 string tensor_format; in FusedBatchNormOpBase() local1390 string tensor_format; in FusedBatchNormGradOpBase() local
650 string tensor_format; in MklFusedBatchNormOp() local1053 string tensor_format; in MklFusedBatchNormGradOp() local
458 Status CheckFormatConstraintsOnShape(const TensorFormat tensor_format, in CheckFormatConstraintsOnShape()
1339 cudnnTensorFormat_t tensor_format; in CheckAndFetchProjectionWeights() local1429 cudnnTensorFormat_t tensor_format; in Create() local