Lines Matching refs:tensor_shape_map
269 RemoteFusedGraphExecuteUtils::TensorShapeMap* tensor_shape_map) {
270 CHECK(tensor_shape_map != nullptr);
310 EmplaceTensorShapeType(name, tensor, tensor_shape_map);
315 EmplaceTensorShapeType(name, tensor, tensor_shape_map);
338 TensorShapeMap* tensor_shape_map) {
339 CHECK_NE(tensor_shape_map, nullptr);
340 tensor_shape_map->clear();
341 tensor_shape_map->reserve(input_node_info_list.size() +
348 EmplaceTensorShapeType(node_name, tensor, tensor_shape_map);
374 const TensorShapeMap& tensor_shape_map, NodeDef* node_def) {
377 auto its = tensor_shape_map.equal_range(node_def->name());
488 TensorShapeMap* tensor_shape_map) {
507 CHECK(tensor_shape_map->count(node_name) == 0);
508 tensor_shape_map->emplace(node_name,
520 const TensorShapeMap& tensor_shape_map, const string& node_name) {
523 return GetTensorShapeType(tensor_shape_map, tid.first.ToString(),
526 return GetTensorShapeType(tensor_shape_map, node_name, 0);
532 const TensorShapeMap& tensor_shape_map, const string& node_name,
535 if (tensor_shape_map.count(node_name) <= 0) {
538 auto its = tensor_shape_map.equal_range(node_name);
567 TensorShapeMap* tensor_shape_map) {
569 CHECK_EQ(tensor_shape_map->count(name), 0);
570 tensor_shape_map->emplace(
579 TensorShapeMap tensor_shape_map;
583 &tensor_shape_map));
593 BuildTensorShapeMapFromGraph(graph, shape_refiner, &tensor_shape_map));
598 AddOutputTensorShapeTypeByTensorShapeMap(tensor_shape_map, &node_def));