Home
last modified time | relevance | path

Searched defs:shape_value (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dconstant_map.cpp100 std::vector<int64_t> shape_value; in GetCompleteShapeInto1DInt64Vector() local
114 auto shape_value = in GetShapeInto1DInt64Vector() local
126 std::vector<int64_t> shape_value; in GetShapeInto1DInt64VectorWithOneUnknown() local
H A Dshape_type_inference.cpp1236 auto shape_value = in ProcessUnsqueezeNode() local
1291 auto shape_value = input_shape.value(); in ComputeConstant() local
2183 c10::SymbolicShape shape_value(final_shape); in ONNXShapeTypeInference() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/
H A Dtf_to_xla_attribute_utils.cc22 Value GetDimValue(OpBuilder &builder, Location loc, Value shape_value, in GetDimValue()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtensor_list_ops_decomposition.cc525 LogicalResult GetConstShapeValue(Value shape_value, in GetConstShapeValue()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/
H A Dcustom_call_encoding.cc176 Value shape_value = ib.create<LLVM::UndefOp>(shape_arr_type); in PackDenseElementsAttribute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dconstant_folding.cc2927 TensorValue shape_value(&tensor); in SimplifyReduction() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
H A Dtfl_ops.cc1458 DenseIntElementsAttr shape_value; in verify() local