Searched refs:InferredShape (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference.h | 29 struct InferredShape { struct
38 typedef std::unordered_map<string, std::vector<InferredShape>> GraphShapeInfo; argument
43 Status InferShapes(Graph* graph, const std::map<int, InferredShape>& arg_shapes,
49 xla::StatusOr<InferredShape> MergeInferredShapes(const InferredShape& a,
50 const InferredShape& b);
|
| D | shape_inference.cc | 46 const std::map<int, InferredShape>& arg_shapes, in PropagateShapes()
83 const InferredShape& arg_shape = it->second; in PropagateShapes()
246 Status InferShapes(Graph* graph, const std::map<int, InferredShape>& arg_shapes, in InferShapes()
273 xla::StatusOr<InferredShape> MergeInferredShapes(const InferredShape& a, in MergeInferredShapes()
274 const InferredShape& b) { in MergeInferredShapes()
275 InferredShape result; in MergeInferredShapes()
|
| D | encapsulate_util.cc | 325 std::map<int, InferredShape> arg_shapes; in PerformStaticShapeInferenceBeforeEncapsulation()
336 [](const InferredShape& inferred_shape) { in PerformStaticShapeInferenceBeforeEncapsulation()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.h | 295 const ParameterInfo& params_info, std::vector<InferredShape>* arg_shapes,
296 std::vector<InferredShape>* retval_shapes);
307 const std::vector<InferredShape>& arg_shapes,
309 const std::vector<InferredShape>& retval_shapes, const Graph& graph,
340 const Node& replicate_node, const std::vector<InferredShape>& arg_shapes,
354 const std::vector<InferredShape>& arg_shapes,
433 const std::vector<InferredShape>& arg_shapes,
|
| D | distributed_tpu_rewrite_pass.cc | 419 const std::vector<InferredShape>& shapes) in TensorDevicePlacer()
478 static int64 GetInferredShapeSize(const InferredShape& ishape, in GetInferredShapeSize()
1680 const InferredShape** info) { in GetEdgeShape()
1694 const ParameterInfo& params_info, std::vector<InferredShape>* arg_shapes, in GetArgAndRetvalShapes()
1695 std::vector<InferredShape>* retval_shapes) { in GetArgAndRetvalShapes()
1711 const InferredShape* info; in GetArgAndRetvalShapes()
1720 xla::StatusOr<InferredShape> status = in GetArgAndRetvalShapes()
1754 const InferredShape* info; in GetArgAndRetvalShapes()
1766 const InferredShape* info; in GetArgAndRetvalShapes()
1768 InferredShape& arg_shape = in GetArgAndRetvalShapes()
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_common.cc | 73 const std::vector<InferredShape>& output_shapes = node_shape_info.second; in ConvertGraphShapeInfoToShapeMap()
462 std::map<int, InferredShape> arg_shapes_for_inference; in RunShapeInferenceOnComputation()
465 InferredShape& shape_for_inference = arg_shapes_for_inference[i]; in RunShapeInferenceOnComputation()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 649 const std::vector<InferredShape>& output_shapes = node_shape_info.second; in GetGraph()
672 const std::vector<InferredShape>& output_shapes = node_shape_info.second; in GetGraph()
|