Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_input_conversion_op.cc58 MklDnnShape input_shape_0; in Compute() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dshape_type_inference.cpp769 auto input_shape_0 = ConstantValueMap::GetShape(n->input(0)->debugName()); in ProcessBroadcastNode() local
873 auto input_shape_0 = in ProcessMatMulNode() local
920 auto input_shape_0 = ConstantValueMap::GetShape(n->input(0)->debugName()); in ProcessReduceNode() local