Searched defs:shape_map (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding.h | 42 const std::unordered_map<string, std::vector<PartialTensorShape>>* shape_map = member
|
| D | graph_optimizer.h | 39 std::unordered_map<string, std::vector<PartialTensorShape>>* shape_map = member
|
| D | constant_folding.cc | 62 const std::unordered_map<string, std::vector<PartialTensorShape>>* in ReadPartialShapesFromShapeMap()
180 const std::unordered_map<string, std::vector<PartialTensorShape>>* in MaybeReplaceShapeOp()
219 const std::unordered_map<string, std::vector<PartialTensorShape>>* in IsConstantFoldable()
|
| D | graph_optimizer.cc | 40 const std::unordered_map<string, std::vector<PartialTensorShape>>* in Optimize()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_common.cc | 63 std::unordered_map<string, std::vector<PartialTensorShape>>* shape_map) { in ConvertGraphShapeInfoToShapeMap()
518 std::unordered_map<string, std::vector<PartialTensorShape>> shape_map; in OptimizeGraph() local
529 std::unordered_map<string, std::vector<PartialTensorShape>> shape_map; in OptimizeGraph() local
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_constants_lib.cc | 255 std::unordered_map<string, std::vector<PartialTensorShape>> shape_map; in FoldConstants() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 646 std::unordered_map<string, std::vector<PartialTensorShape>> shape_map; in GetGraph() local
669 std::unordered_map<string, std::vector<PartialTensorShape>> shape_map; in GetGraph() local
|