Home
last modified time | relevance | path

Searched refs:FusedBatchNormShape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dmkl_nn_ops.cc1760 .SetShapeFn(shape_inference::FusedBatchNormShape)
1831 .SetShapeFn(shape_inference::FusedBatchNormShape)
1856 .SetShapeFn(shape_inference::FusedBatchNormShape)
1907 .SetShapeFn(shape_inference::FusedBatchNormShape);
1945 .SetShapeFn(shape_inference::FusedBatchNormShape)
1995 .SetShapeFn(shape_inference::FusedBatchNormShape)
Dnn_ops.cc181 .SetShapeFn(shape_inference::FusedBatchNormShape);
200 .SetShapeFn(shape_inference::FusedBatchNormShape);
2546 .SetShapeFn(shape_inference::FusedBatchNormShape)
2616 .SetShapeFn(shape_inference::FusedBatchNormShape);
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.h154 Status FusedBatchNormShape(shape_inference::InferenceContext* c);
Dcommon_shape_fns.cc1131 Status FusedBatchNormShape(shape_inference::InferenceContext* c) { in FusedBatchNormShape() function
1174 TF_RETURN_IF_ERROR(FusedBatchNormShape(c)); in FusedBatchNormV3Shape()