Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dshape_type_inference.cpp581 auto input_shape_value = input_shape[input_shape_idx].static_size(); in ComputeShapeFromExpand() local
798 auto input_shape_value = input_shape.value().sizes(); in ProcessShapeForConcatNode() local
820 auto input_shape_value = input_shape.value().sizes(); in ProcessShapeForConcatNode() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/
H A Dtf_to_xla_attribute_utils.cc101 Value input_shape_value = builder.create<TF::ShapeOp>( in PadForDynamicShapedInputSamePadding() local