Searched refs:shape_array (Results 1 – 6 of 6) sorted by relevance
64 Array& shape_array = model->GetOrCreateArray(shape_array_name); in Run() local65 *(shape_array.mutable_shape()->mutable_dims()) = { in Run()68 shape_array.data_type = ArrayDataType::kInt32; in Run()69 auto& shape_buffer = shape_array.GetMutableBuffer<ArrayDataType::kInt32>(); in Run()
75 Array& shape_array = model->GetOrCreateArray(shape_array_name); in Run() local76 *(shape_array.mutable_shape()->mutable_dims()) = { in Run()79 shape_array.data_type = ArrayDataType::kInt32; in Run()80 auto& shape_buffer = shape_array.GetMutableBuffer<ArrayDataType::kInt32>(); in Run()
95 Array& shape_array = model->GetOrCreateArray(shape_array_name); in Run() local96 *(shape_array.mutable_shape()->mutable_dims()) = { in Run()99 shape_array.data_type = ArrayDataType::kInt32; in Run()100 auto& shape_buffer = shape_array.GetMutableBuffer<ArrayDataType::kInt32>(); in Run()
410 auto& shape_array = model->GetArray(op->inputs[1]); in ProcessTensorFlowReshapeOperator() local411 if (!shape_array.has_shape()) { in ProcessTensorFlowReshapeOperator()415 if (!shape_array.buffer) { in ProcessTensorFlowReshapeOperator()419 CHECK(shape_array.data_type == ArrayDataType::kInt32) in ProcessTensorFlowReshapeOperator()428 shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessTensorFlowReshapeOperator()
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()
1052 const auto& shape_array = model.GetArray(src_op.inputs[1]); in ConvertTensorFlowReshapeOperator() local1053 QCHECK(shape_array.data_type == ArrayDataType::kInt32) in ConvertTensorFlowReshapeOperator()1055 QCHECK(shape_array.buffer != nullptr) in ConvertTensorFlowReshapeOperator()1057 const auto& shape_data = shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ConvertTensorFlowReshapeOperator()