| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| H A D | linear_operator_circulant_test.py | 293 shape_info, argument
378 shape_info, argument
452 shape_info, argument
714 shape_info, argument
773 shape_info, argument
|
| H A D | linear_operator_block_diag_test.py | 102 self, shape_info, dtype, use_placeholder, argument
476 self, shape_info, dtype, use_placeholder, argument
|
| H A D | linear_operator_block_lower_triangular_test.py | 100 self, shape_info, dtype, use_placeholder, argument
|
| H A D | linear_operator_low_rank_update_test.py | 71 def operator_and_matrix(self, shape_info, dtype, use_placeholder, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A 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
|
| H A D | shape_inference.cc | 207 GraphShapeInfo* shape_info) { in StoreOutputShapes()
245 GraphShapeInfo* shape_info) { in InferShapes()
|
| H A D | test_util.cc | 24 const Graph& graph, const GraphShapeInfo& shape_info, in ShapeAnnotationsMatch()
|
| H A D | encapsulate_util.cc | 324 GraphShapeInfo shape_info; in PerformStaticShapeInferenceBeforeEncapsulation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | tpu_compile.cc | 189 GraphShapeInfo shape_info; in OptimizeGraph() local
202 GraphShapeInfo shape_info; in OptimizeGraph() local
341 FunctionLibraryRuntime* flr, GraphShapeInfo* shape_info) { in RunShapeInferenceOnComputation()
|
| /aosp_15_r20/external/pytorch/test/cpp/tensorexpr/ |
| H A D | test_expr.cpp | 137 auto shape_info = shape_gen_fn(dims[i], channels_last_cont_shape_conf); in TEST() local
146 auto shape_info = shape_gen_fn(dims[i], channels_last_non_cont_shape_conf); in TEST() local
155 auto shape_info = shape_gen_fn(dims[i], cont_shape_conf); in TEST() local
164 auto shape_info = shape_gen_fn(dims[i], channels_last_cont_shape_conf); in TEST() local
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A D | NERNNLayer.cpp | 63 …auto shape_info = TensorInfo(misc::shape_calculator::compute_rnn_shape(recurrent_weights, hidden_s… in validate() local
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
| H A D | CLRNNLayer.cpp | 66 …auto shape_info = TensorInfo(compute_rnn_shape(recurrent_weights, hidden_state->dimension(idx_heig… in validate() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_compiler.cc | 646 GraphShapeInfo shape_info; in GetGraph() local
669 GraphShapeInfo shape_info; in GetGraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_to_tensor_op_test.py | 694 def wrap_in_placeholder(self, arg, shape_info): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| H A D | conv_emitter.cc | 72 ShapeInfo shape_info; in GetShapeInfo() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | light_outside_compilation.cc | 378 std::vector<int32_t> shape_info(num_dimensions); in PopulateMetadataBufferIfNeeded() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | distributed_tpu_rewrite_pass.cc | 1892 static Status GetEdgeShape(const GraphShapeInfo& shape_info, const Edge& edge, in GetEdgeShape()
1906 const GraphShapeInfo& shape_info, const Node& node, in GetArgAndRetvalShapes()
4717 const GraphShapeInfo& shape_info, in RewriteTPUReplicateNode()
4996 GraphShapeInfo shape_info; in InternalRun() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| H A D | quantization_ops_test.cc | 374 GraphShapeInfo shape_info; in ConvertAndRun() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_functional_ops.cc | 711 GraphShapeInfo* shape_info) { in GroupTensorsForOutputPacking()
|