Searched refs:engine_input_shapes (Results 1 – 2 of 2) sorted by relevance
129 std::vector<TensorShape> engine_input_shapes; in Compute() local131 engine_input_shapes.emplace_back(shape); in Compute()157 resource->cache_.emplace(engine_input_shapes, in Compute()214 const std::vector<TensorShape>& engine_input_shapes = pair.first; in Compute() local215 for (const TensorShape& shape : engine_input_shapes) { in Compute()
930 std::vector<TensorShape> engine_input_shapes(input_concrete_shapes); in GetEngine() local931 for (int i = 0; i < engine_input_shapes.size(); i++) { in GetEngine()932 engine_input_shapes[i].set_dim(0, max_batch_size); in GetEngine()942 cache.emplace(engine_input_shapes, in GetEngine()955 return std::pair<EngineContext*, int>(cache.at(engine_input_shapes).get(), in GetEngine()