Searched defs:nested_size (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorFactories.cpp | 49 auto nested_size = self_nt->get_nested_sizes().clone(); in empty_like_nested() local
63 auto nested_size = self_nt->get_nested_sizes().clone(); in empty_like_nested() local
|
| H A D | NestedTensorMath.cpp | 903 std::tuple<Tensor, Tensor> _nested_compute_contiguous_strides_offsets(const Tensor& nested_size) { in _nested_compute_contiguous_strides_offsets()
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | input_metadata.cpp | 12 auto nested_size = input._nested_tensor_size(); in compute_variant_shape() local
|
| /aosp_15_r20/prebuilts/runtime/mainline/platform/sdk/include_gen/external/perfetto/perfetto_protos_perfetto_trace_non_minimal_lite_gen_headers/gen/external/perfetto/protos/perfetto/trace/ |
| H A D | test_event.pb.h | 642 inline int TestEvent_TestPayload::nested_size() const { in nested_size() function
|
| /aosp_15_r20/out/soong/.intermediates/external/perfetto/perfetto_protos_perfetto_trace_non_minimal_lite_gen_headers/gen/external/perfetto/protos/perfetto/trace/ |
| D | test_event.pb.h | 642 inline int TestEvent_TestPayload::nested_size() const { in nested_size() function
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_nestedtensor.py | 2934 def grad_test_func(tensor, nested_size): argument
2950 def grad_test_func(tensor, nested_size): argument
3523 def _get_list_for_jagged_tensor(self, nested_size, device, requires_grad=True): argument
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_subclasses.py | 45 def get_jagged_tensor(nested_size, offsets, requires_grad=True): argument
1807 def _get_jagged_tensor(self, nested_size, offsets, requires_grad=True): argument
|
| /aosp_15_r20/external/linux-kselftest/tools/testing/selftests/kvm/lib/x86_64/ |
| H A D | processor.c | 964 static int nested_size = -1; in vcpu_save_state() local
|
| /aosp_15_r20/external/rust/android-crates-io/crates/grpcio-sys/grpc/tools/codegen/core/ |
| D | gen_huffman_decompressor.cc | 1104 size_t nested_size = nested->Size(); in Choose() local
|
| /aosp_15_r20/external/grpc-grpc/tools/codegen/core/ |
| H A D | gen_huffman_decompressor.cc | 1156 size_t nested_size = nested->Size(); in Choose() local
|