Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dnn_ops.cc665 .SetShapeFn(shape_inference::DepthwiseConv2DNativeShape);
3334 .SetShapeFn(shape_inference::DepthwiseConv2DNativeShape);
3353 .SetShapeFn(shape_inference::DepthwiseConv2DNativeShape);
3373 .SetShapeFn(shape_inference::DepthwiseConv2DNativeShape);
3396 .SetShapeFn(shape_inference::DepthwiseConv2DNativeShape);
H A Dmkl_nn_ops.cc328 .SetShapeFn(shape_inference::DepthwiseConv2DNativeShape);
348 .SetShapeFn(shape_inference::DepthwiseConv2DNativeShape);
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcommon_shape_fns.h142 Status DepthwiseConv2DNativeShape(shape_inference::InferenceContext* c);
H A Dcommon_shape_fns.cc1100 Status DepthwiseConv2DNativeShape(shape_inference::InferenceContext* c) { in DepthwiseConv2DNativeShape() function