Searched defs:tensor_format (Results 1 – 5 of 5) sorted by relevance
436 TensorFormat tensor_format, char dimension) { in GetTensorDim()468 TensorFormat tensor_format, char dimension) { in GetTensorDim()484 inline int64 GetTensorDim(const Tensor& tensor, TensorFormat tensor_format, in GetTensorDim()498 const std::vector<int64>& explicit_paddings, TensorFormat tensor_format, in GetExplicitPaddingForDim()
1106 string tensor_format; in FusedBatchNormOpBase() local1283 string tensor_format; in FusedBatchNormGradOpBase() local
522 string tensor_format; in MklFusedBatchNormOp() local861 string tensor_format; in MklFusedBatchNormGradOp() local
567 Status CheckFormatConstraintsOnShape(const TensorFormat tensor_format, in CheckFormatConstraintsOnShape()
1298 cudnnTensorFormat_t tensor_format; in CheckAndFetchProjectionWeights() local1388 cudnnTensorFormat_t tensor_format; in Create() local