| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference_test.cc | 51 GraphShapeInfo shape_info; in TEST() local
115 GraphShapeInfo shape_info; in TEST() local
161 GraphShapeInfo shape_info; in TEST() local
|
| D | test_util.cc | 24 const Graph& graph, const GraphShapeInfo& shape_info, in ShapeAnnotationsMatch()
|
| D | shape_inference.cc | 210 GraphShapeInfo* shape_info) { in StoreOutputShapes()
248 GraphShapeInfo* shape_info) { in InferShapes()
|
| D | encapsulate_util.cc | 326 GraphShapeInfo shape_info; in PerformStaticShapeInferenceBeforeEncapsulation() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_circulant_test.py | 118 shape_info, argument
180 shape_info, argument
265 shape_info, argument
501 shape_info, argument
578 shape_info, argument
|
| D | linear_operator_block_lower_triangular_test.py | 98 self, shape_info, dtype, use_placeholder, argument
|
| D | linear_operator_block_diag_test.py | 91 self, shape_info, dtype, use_placeholder, argument
|
| D | linear_operator_low_rank_update_test.py | 74 def operator_and_matrix(self, shape_info, dtype, use_placeholder, argument
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_common.cc | 458 FunctionLibraryRuntime* flr, GraphShapeInfo* shape_info) { in RunShapeInferenceOnComputation()
513 GraphShapeInfo shape_info; in OptimizeGraph() local
526 GraphShapeInfo shape_info; in OptimizeGraph() local
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_to_tensor_op_test.py | 690 def wrap_in_placeholder(self, arg, shape_info): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter.cc | 70 ShapeInfo shape_info; in GetShapeInfo() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 641 GraphShapeInfo shape_info; in GetGraph() local
664 GraphShapeInfo shape_info; in GetGraph() local
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 1679 static Status GetEdgeShape(const GraphShapeInfo& shape_info, const Edge& edge, in GetEdgeShape()
1693 const GraphShapeInfo& shape_info, const Node& node, in GetArgAndRetvalShapes()
4216 const GraphShapeInfo& shape_info, in RewriteTPUReplicateNode()
4470 GraphShapeInfo shape_info; in Run() local
|