Searched defs:in_shapes (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_device_analysis.py | 42 def prop_device_on_graph(self, graph, example_devices, in_shapes=None): argument
61 self, fn, in_devices, expected_device, in_shapes=None, subtest_str="" argument
|
| H A D | test_dtype_analysis.py | 127 def assert_dtype_equal(self, fn, in_shapes, in_dtypes): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | batch_ops.cc | 78 std::vector<shape_inference::ShapeHandle> in_shapes; in __anoncf35e64b0102() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | shape_inference_testutil.cc | 38 std::vector<ShapeHandle> in_shapes; in InferShapes() local
|