Home
last modified time | relevance | path

Searched defs:input_shape_1 (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_input_conversion_op.cc62 MklDnnShape input_shape_1; in Compute() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dshape_type_inference.cpp771 auto input_shape_1 = ConstantValueMap::GetShape(n->input(1)->debugName()); in ProcessBroadcastNode() local
876 auto input_shape_1 = in ProcessMatMulNode() local