Searched defs:shape_map (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/federated-compute/fcp/aggregation/tensorflow/ |
| H A D | checkpoint_reader.cc | 62 TensorShapeMap shape_map; in Create() local
76 DataTypeMap data_type_map, TensorShapeMap shape_map) in CheckpointReader()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | tpu_compile.cc | 137 std::unordered_map<string, std::vector<PartialTensorShape>>* shape_map) { in ConvertGraphShapeInfoToShapeMap()
194 std::unordered_map<string, std::vector<PartialTensorShape>> shape_map; in OptimizeGraph() local
205 std::unordered_map<string, std::vector<PartialTensorShape>> shape_map; in OptimizeGraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | constant_folding.cc | 63 const std::unordered_map<string, std::vector<PartialTensorShape>>* in ReadPartialShapesFromShapeMap()
181 const std::unordered_map<string, std::vector<PartialTensorShape>>* in MaybeReplaceShapeOp()
220 const std::unordered_map<string, std::vector<PartialTensorShape>>* in IsConstantFoldable()
|
| H A D | constant_folding.h | 42 const std::unordered_map<string, std::vector<PartialTensorShape>>* shape_map = member
|
| H A D | graph_optimizer.h | 39 std::unordered_map<string, std::vector<PartialTensorShape>>* shape_map = member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_compiler.cc | 651 std::unordered_map<string, std::vector<PartialTensorShape>> shape_map; in GetGraph() local
674 std::unordered_map<string, std::vector<PartialTensorShape>> shape_map; in GetGraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | fold_constants_lib.cc | 255 std::unordered_map<string, std::vector<PartialTensorShape>> shape_map; in FoldConstants() local
|