Searched refs:dims_array (Results 1 – 3 of 3) sorted by relevance
539 const auto dims_array = tfl_tensor->dims; in ValidateOutputTensorShapeConsistency() local540 std::vector<DimensionHandle> dims(dims_array->size); in ValidateOutputTensorShapeConsistency()541 for (int j = 0; j < dims_array->size; ++j) { in ValidateOutputTensorShapeConsistency()542 dims[j] = c.MakeDim(dims_array->data[j]); in ValidateOutputTensorShapeConsistency()
351 auto& dims_array = model->GetArray(op->inputs[0]); in ProcessOpWithShapeInput() local352 if (!dims_array.has_shape()) { in ProcessOpWithShapeInput()356 if (!dims_array.buffer) { in ProcessOpWithShapeInput()360 CHECK(dims_array.data_type == ArrayDataType::kInt32) << "dims must be int32"; in ProcessOpWithShapeInput()361 CHECK_LE(RequiredBufferSizeForShape(dims_array.shape()), 6) in ProcessOpWithShapeInput()365 dims_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessOpWithShapeInput()
2729 std::vector<int32> dims_array(trt_input_rank + 1, 1); in TEST_P() local2731 dims_array[1] = (data_format == "NHWC" ? 3 : 2); in TEST_P()2733 dims_array[1] = 2; in TEST_P()2734 dims_array[trt_input_rank] = 3; in TEST_P()2736 const int num_input = TrtTensorDimsNumElements(GetTestDims(dims_array)); in TEST_P()2741 AddTestTensor("input", dims_array, input_data); in TEST_P()2766 TestOpConverter("my_biasadd", node_def, dims_array, Status::OK(), in TEST_P()