Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dmkl_remapper_test.cc846 auto mul_shape = ops::Placeholder::Shape({64, 64}); in RunTest() local
H A Dconstant_folding.cc3558 const auto& mul_shape = mul_props[0].shape(); in MulConvPushDown() local
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/
H A Dpass.cc2926 ShapedType mul_shape = (*op->result_type_begin()).cast<ShapedType>(); in matchAndRewrite() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLinearAlgebra.cpp3389 c10::SmallVector<int64_t, 10> mul_shape(2 * maxdim); in kron_out() local