Searched refs:shape_map (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_optimizer.cc | 40 shape_map, in Optimize()
64 cf_opts.shape_map = shape_map; in Optimize()
114 Optimize(runtime, env, device, graph, options.shape_map, in Optimize()
|
| D | constant_folding.cc | 61 shape_map, in ReadPartialShapesFromShapeMap()
63 CHECK(shape_map != nullptr); in ReadPartialShapesFromShapeMap()
68 const auto known_shape_iter = shape_map->find(in->src()->name()); in ReadPartialShapesFromShapeMap()
69 if (known_shape_iter == shape_map->end()) { in ReadPartialShapesFromShapeMap()
179 shape_map, in MaybeReplaceShapeOp()
182 if (shape_map == nullptr || !IsShapeOp(n)) { in MaybeReplaceShapeOp()
187 if (!ReadPartialShapesFromShapeMap(n, shape_map, &input_shapes)) { in MaybeReplaceShapeOp()
218 shape_map, in IsConstantFoldable()
225 if (MaybeReplaceShapeOp(n, shape_map, shape_replacement_map)) { in IsConstantFoldable()
271 if (IsConstantFoldable(n, opts.shape_map, opts.consider, in ConsiderConstantFoldableNode()
|
| D | graph_optimizer.h | 39 std::unordered_map<string, std::vector<PartialTensorShape>>* shape_map = member
67 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/contrib/checkpoint/python/ |
| D | visualize.py | 55 shape_map = reader.get_variable_to_shape_map()
80 shape_map[attribute.checkpoint_key],
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| D | ops.py | 562 def random_crop(labeled_tensor, shape_map, seed=None, name=None): argument
584 for axis_name in shape_map:
592 if axis.name in shape_map:
593 size = shape_map[axis.name]
|