Searched defs:check_shape (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_pytorch_jit_onnx.py | 52 check_shape = True variable in _TestJITIRToONNX
|
| H A D | onnx_test_common.py | 131 check_shape = True variable in _TestONNXRuntime
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/structured/ |
| H A D | structured_array_ops_test.py | 68 def _assertStructuredEqual(self, a, b, msg, check_shape): argument
|
| H A D | structured_tensor_test.py | 94 def _assertStructuredEqual(self, a, b, msg, check_shape): argument
|
| /aosp_15_r20/external/pytorch/torch/export/ |
| H A D | dynamic_shapes.py | 751 def check_shape(path, t, dynamic_shape): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/autograph/operators/ |
| H A D | control_flow.py | 210 name, check_shape, init, entry, exit_, shape_invariant): argument
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | verification.py | 81 check_shape: bool = True variable in VerificationOptions
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ |
| H A D | flatbuffer_export.cc | 879 auto check_shape = in BuildTensor() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | distributed_tpu_rewrite_pass.cc | 1923 auto check_shape = [&](int input_index) -> Status { in GetArgAndRetvalShapes() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_linalg.py | 3228 def check_shape(a_shape, ind): function
|