Searched refs:shape_array (Results 1 – 8 of 8) sorted by relevance
64 Array& shape_array = model->GetOrCreateArray(shape_array_name); in Run() local66 *(shape_array.mutable_shape()->mutable_dims()) = {shape_array_dims}; in Run()68 shape_array.data_type = ArrayDataType::kInt32; in Run()69 auto& shape_buffer = shape_array.GetMutableBuffer<ArrayDataType::kInt32>(); in Run()
82 Array& shape_array = model->GetOrCreateArray(shape_array_name); in Run() local83 *(shape_array.mutable_shape()->mutable_dims()) = { in Run()86 shape_array.data_type = ArrayDataType::kInt32; in Run()87 auto& shape_buffer = shape_array.GetMutableBuffer<ArrayDataType::kInt32>(); in Run()
96 Array& shape_array = model->GetOrCreateArray(shape_array_name); in Run() local97 *(shape_array.mutable_shape()->mutable_dims()) = { in Run()100 shape_array.data_type = ArrayDataType::kInt32; in Run()101 auto& shape_buffer = shape_array.GetMutableBuffer<ArrayDataType::kInt32>(); in Run()
196 Array& shape_array = model->GetOrCreateArray(shape_array_name); in Run() local198 *(shape_array.mutable_shape()->mutable_dims()) = {dim_size}; in Run()199 shape_array.data_type = ArrayDataType::kInt32; in Run()200 auto& shape_buffer = shape_array.GetMutableBuffer<ArrayDataType::kInt32>(); in Run()
411 auto& shape_array = model->GetArray(op->inputs[1]); in ProcessTensorFlowReshapeOperator() local412 if (!shape_array.has_shape()) { in ProcessTensorFlowReshapeOperator()416 if (!shape_array.buffer) { in ProcessTensorFlowReshapeOperator()420 CHECK(shape_array.data_type == ArrayDataType::kInt32) in ProcessTensorFlowReshapeOperator()429 shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessTensorFlowReshapeOperator()2126 auto& shape_array = model->GetArray(op->inputs[2]); in ProcessScatterNdOperator() local2129 if (!shape_array.has_shape()) { in ProcessScatterNdOperator()2133 if (!shape_array.buffer) { in ProcessScatterNdOperator()2137 CHECK(shape_array.data_type == ArrayDataType::kInt32) << "dims must be int32"; in ProcessScatterNdOperator()2140 shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessScatterNdOperator()
475 std::array<int64, SHAPE_ARRAY_SIZE> shape_array; in RegisterConstantNode() local481 shape_array = BuildShapeArray(shape_handle, context); in RegisterConstantNode()489 const_node_info.add_shape(shape_array[0]); in RegisterConstantNode()490 const_node_info.add_shape(shape_array[1]); in RegisterConstantNode()491 const_node_info.add_shape(shape_array[2]); in RegisterConstantNode()492 const_node_info.add_shape(shape_array[3]); in RegisterConstantNode()618 std::array<int64, SHAPE_ARRAY_SIZE> shape_array; in IsNodeFlattenReshape() local624 shape_array = BuildShapeArray(shape_handle, context); in IsNodeFlattenReshape()632 shape_array = ToTensorShapeArray(shapes.at(0)); in IsNodeFlattenReshape()636 if (shape_array[0] == 1 && shape_array[1] == 1 && shape_array[2] == 1) { in IsNodeFlattenReshape()
103 int shape_array[2] = {segmentation.height(), segmentation.width()}; in ConvertToSegmentationResults() local104 env->SetIntArrayRegion(jmask_shape, 0, 2, shape_array); in ConvertToSegmentationResults()
1060 const auto& shape_array = model.GetArray(src_op.inputs[1]); in ConvertTensorFlowReshapeOperator() local1061 QCHECK(shape_array.data_type == ArrayDataType::kInt32) in ConvertTensorFlowReshapeOperator()1063 QCHECK(shape_array.buffer != nullptr) in ConvertTensorFlowReshapeOperator()1065 const auto& shape_data = shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ConvertTensorFlowReshapeOperator()