Searched refs:shape_map (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding.cc | 63 shape_map, in ReadPartialShapesFromShapeMap()
65 CHECK(shape_map != nullptr); in ReadPartialShapesFromShapeMap()
70 const auto known_shape_iter = shape_map->find(in->src()->name()); in ReadPartialShapesFromShapeMap()
71 if (known_shape_iter == shape_map->end()) { in ReadPartialShapesFromShapeMap()
181 shape_map, in MaybeReplaceShapeOp()
184 if (shape_map == nullptr || !IsShapeOp(n)) { in MaybeReplaceShapeOp()
189 if (!ReadPartialShapesFromShapeMap(n, shape_map, &input_shapes)) { in MaybeReplaceShapeOp()
220 shape_map, in IsConstantFoldable()
228 if (MaybeReplaceShapeOp(n, shape_map, shape_replacement_map)) { in IsConstantFoldable()
237 if (shape_map != nullptr) { in IsConstantFoldable()
[all …]
|
| D | graph_optimizer.cc | 41 shape_map, in Optimize()
68 cf_opts.shape_map = shape_map; in Optimize()
147 runtime, env, device, graph, options.shape_map, options.cse_consider_fn, in Optimize()
|
| D | graph_optimizer.h | 39 std::unordered_map<string, std::vector<PartialTensorShape>>* shape_map = member
82 shape_map,
|
| D | constant_folding.h | 42 const std::unordered_map<string, std::vector<PartialTensorShape>>* shape_map = member
|
| D | constant_folding_test.cc | 482 opts.shape_map = ↦ in TEST_F()
559 opts.shape_map = ↦ in TEST_F()
613 opts.shape_map = ↦ in TEST_F()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_constants_lib.cc | 255 std::unordered_map<string, std::vector<PartialTensorShape>> shape_map; in FoldConstants() local
262 std::vector<PartialTensorShape>& partial_shapes = shape_map[node->name()]; in FoldConstants()
283 auto& recv_partial_shapes = shape_map[recv_name]; in FoldConstants()
298 cf_opts.shape_map = &shape_map; in FoldConstants()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_common.cc | 63 std::unordered_map<string, std::vector<PartialTensorShape>>* shape_map) { in ConvertGraphShapeInfoToShapeMap() argument
80 auto& partial_shapes = (*shape_map)[node_name]; in ConvertGraphShapeInfoToShapeMap()
518 std::unordered_map<string, std::vector<PartialTensorShape>> shape_map; in OptimizeGraph() local
519 ConvertGraphShapeInfoToShapeMap(**graph, shape_info, &shape_map); in OptimizeGraph()
520 optimizer_opts.shape_map = &shape_map; in OptimizeGraph()
529 std::unordered_map<string, std::vector<PartialTensorShape>> shape_map; in OptimizeGraph() local
530 ConvertGraphShapeInfoToShapeMap(**graph, shape_info, &shape_map); in OptimizeGraph()
531 optimizer.Optimize(flr, flr->env(), flr->device(), graph, &shape_map); in OptimizeGraph()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 646 std::unordered_map<string, std::vector<PartialTensorShape>> shape_map; in GetGraph() local
652 auto& partial_shapes = shape_map[node_name]; in GetGraph()
658 graph_optimizer_options.shape_map = &shape_map; in GetGraph()
669 std::unordered_map<string, std::vector<PartialTensorShape>> shape_map; in GetGraph() local
675 auto& partial_shapes = shape_map[node_name]; in GetGraph()
681 graph_optimizer_options.shape_map = &shape_map; in GetGraph()
|