Home
last modified time | relevance | path

Searched full:nested_size (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/perfetto/src/tracing/
H A Dtraced_proto_unittest.cc320 EXPECT_EQ(result.nested_size(), 2); in TEST_F()
333 EXPECT_EQ(result.nested_size(), 2); in TEST_F()
349 EXPECT_EQ(result.nested_size(), 2); in TEST_F()
361 EXPECT_EQ(result.nested_size(), 1); in TEST_F()
372 EXPECT_EQ(result.nested_size(), 1); in TEST_F()
385 EXPECT_EQ(result.nested_size(), 2); in TEST_F()
397 EXPECT_EQ(result.nested_size(), 2); in TEST_F()
409 EXPECT_EQ(result.nested_size(), 2); in TEST_F()
425 EXPECT_EQ(result.nested_size(), 2); in TEST_F()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorFactories.cpp49 auto nested_size = self_nt->get_nested_sizes().clone(); in empty_like_nested() local
50 int64_t buffer_size = get_numel_from_nested_size_tensor(nested_size); in empty_like_nested()
52 auto tensor = wrap_buffer(new_buffer, nested_size); in empty_like_nested()
63 auto nested_size = self_nt->get_nested_sizes().clone(); in empty_like_nested() local
66 auto tensor = wrap_buffer(new_buffer, nested_size, nested_strides, offsets); in empty_like_nested()
H A DNestedTensorMath.cpp903 std::tuple<Tensor, Tensor> _nested_compute_contiguous_strides_offsets(const Tensor& nested_size) { in _nested_compute_contiguous_strides_offsets() argument
905 construct_nested_strides(nested_size), in _nested_compute_contiguous_strides_offsets()
906 construct_offsets(nested_size)); in _nested_compute_contiguous_strides_offsets()
/aosp_15_r20/external/linux-kselftest/tools/testing/selftests/kvm/lib/x86_64/
H A Dprocessor.c964 static int nested_size = -1; in vcpu_save_state() local
966 if (nested_size == -1) { in vcpu_save_state()
967 nested_size = kvm_check_cap(KVM_CAP_NESTED_STATE); in vcpu_save_state()
968 TEST_ASSERT(nested_size <= sizeof(state->nested_), in vcpu_save_state()
970 nested_size, sizeof(state->nested_)); in vcpu_save_state()
993 if (nested_size) { in vcpu_save_state()
997 TEST_ASSERT(state->nested.size <= nested_size, in vcpu_save_state()
999 state->nested.size, nested_size); in vcpu_save_state()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dinput_metadata.cpp12 auto nested_size = input._nested_tensor_size(); in compute_variant_shape() local
13 return MetadataShape{std::in_place_type<at::Tensor>, nested_size}; in compute_variant_shape()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DNestedTensorImpl.h17 at::Tensor construct_nested_strides(const at::Tensor& nested_size);
18 at::Tensor construct_offsets(const at::Tensor& nested_size);
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_subclasses.py45 def get_jagged_tensor(nested_size, offsets, requires_grad=True): argument
48 D = nested_size[1]
50 for s in nested_size[0]:
1807 def _get_jagged_tensor(self, nested_size, offsets, requires_grad=True): argument
1808 return get_jagged_tensor(nested_size, offsets, requires_grad)
/aosp_15_r20/external/pytorch/test/
H A Dtest_nestedtensor.py435 # self.assertEqual(default_nested_tensor.nested_size(), ())
2929 nested_size = torch.tensor([[1, 2], [2, 2]])
2934 def grad_test_func(tensor, nested_size): argument
2936 tensor, nested_size, fuse_transform_0213=False
2941 data = (padded_tensor, nested_size)
2945 nested_size = torch.tensor([[1, 8], [2, 8]])
2950 def grad_test_func(tensor, nested_size): argument
2952 tensor, nested_size, fuse_transform_0213=True
2957 data = (padded_tensor, nested_size)
3523 def _get_list_for_jagged_tensor(self, nested_size, device, requires_grad=True): argument
[all …]
/aosp_15_r20/external/rust/android-crates-io/crates/grpcio-sys/grpc/tools/codegen/core/
Dgen_huffman_decompressor.cc1104 size_t nested_size = nested->Size(); in Choose() local
1109 if (nested_size < best_size) { in Choose()
1111 best_size = nested_size; in Choose()
/aosp_15_r20/external/grpc-grpc/tools/codegen/core/
H A Dgen_huffman_decompressor.cc1156 size_t nested_size = nested->Size(); in Choose() local
1161 if (nested_size < best_size) { in Choose()
1163 best_size = nested_size; in Choose()
/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 Dtest_event.pb.h212 int nested_size() const; in Swap()
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/
Dtest_event.pb.h212 int nested_size() const; in Swap()
642 inline int TestEvent_TestPayload::nested_size() const { in nested_size() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerFunctions.cu384 T* output, // [batch_size x max(input.nested_size(1)) x inner_size] in add_padding_kernelLauncher()
/aosp_15_r20/external/pytorch/tools/autograd/
H A Dderivatives.yaml2792 - name: _nested_view_from_buffer(Tensor(a) self, Tensor nested_size, Tensor nested_strides, Tensor…
2794 nested_size: non_differentiable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dnative_functions.yaml6174 - func: _nested_view_from_buffer(Tensor(a) self, Tensor nested_size, Tensor nested_strides, Tensor …
6180 - func: _nested_view_from_buffer_copy(Tensor self, Tensor nested_size, Tensor nested_strides, Tenso…
6234 - func: _nested_compute_contiguous_strides_offsets(Tensor nested_size) -> (Tensor, Tensor)