Searched refs:conversion_input_shapes (Results 1 – 1 of 1) sorted by relevance
1053 const std::vector<PartialTensorShape>& conversion_input_shapes = in BuildEngine() local1060 << " with input shapes: " << DebugString(conversion_input_shapes); in BuildEngine()1072 conversion_input_shapes, &logger, cache_resource->allocator_.get(), in BuildEngine()1316 const std::vector<PartialTensorShape>& conversion_input_shapes = in AllocateCalibrationResources() local1323 cres->thr_.reset(new std::thread([this, cres, shapes, conversion_input_shapes, in AllocateCalibrationResources()1355 conversion_input_shapes, &cache_res->GetLogger(), in AllocateCalibrationResources()