Home
last modified time | relevance | path

Searched defs:conv_shape (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtranspose_folding_test.cc249 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F() local
307 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F() local
370 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F() local
439 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
H A Ddepthwise_conv_plus_1x1_conv.cc263 const auto conv_shape = conv_attr->weights.shape; in IsNodeSupported() local
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/
H A Dpass.cc2927 ShapedType conv_shape = in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dconstant_folding.cc3559 const auto& conv_shape = conv_props[0].shape(); in MulConvPushDown() local