Searched refs:shape_array (Results 1 – 9 of 9) 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()
97 Array& shape_array = model->GetOrCreateArray(shape_array_name); in Run() local98 *(shape_array.mutable_shape()->mutable_dims()) = { in Run()101 shape_array.data_type = ArrayDataType::kInt32; in Run()102 auto& shape_buffer = shape_array.GetMutableBuffer<ArrayDataType::kInt32>(); in Run()
197 Array& shape_array = model->GetOrCreateArray(shape_array_name); in Run() local199 *(shape_array.mutable_shape()->mutable_dims()) = {dim_size}; in Run()200 shape_array.data_type = ArrayDataType::kInt32; in Run()201 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()2127 auto& shape_array = model->GetArray(op->inputs[2]); in ProcessScatterNdOperator() local2130 if (!shape_array.has_shape()) { in ProcessScatterNdOperator()2134 if (!shape_array.buffer) { in ProcessScatterNdOperator()2138 CHECK(shape_array.data_type == ArrayDataType::kInt32) << "dims must be int32"; in ProcessScatterNdOperator()2141 shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessScatterNdOperator()
79 int32_t *shape_array = new int32_t[num_dims]; in create_tensor_shape_array() local83 shape_array[d] = info.tensor_shape()[d]; in create_tensor_shape_array()86 return shape_array; in create_tensor_shape_array()
406 std::vector<std::string> shape_array; in convertIValue() local412 shape_array.push_back(std::get<0>(tuple)); in convertIValue()420 vectorToString(shape_array), in convertIValue()427 std::vector<std::string> shape_array; in convertIValue() local433 shape_array.push_back(std::get<0>(tuple)); in convertIValue()446 vectorToString(shape_array), in convertIValue()
103 int shape_array[2] = {segmentation.height(), segmentation.width()}; in ConvertToSegmentationResults() local104 env->SetIntArrayRegion(jmask_shape, 0, 2, shape_array); in ConvertToSegmentationResults()
1065 const auto& shape_array = model.GetArray(src_op.inputs[1]); in ConvertTensorFlowReshapeOperator() local1066 QCHECK(shape_array.data_type == ArrayDataType::kInt32) in ConvertTensorFlowReshapeOperator()1068 QCHECK(shape_array.buffer != nullptr) in ConvertTensorFlowReshapeOperator()1070 const auto& shape_data = shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ConvertTensorFlowReshapeOperator()