Searched defs:test_elements (Results 1 – 9 of 9) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | TensorCompare.cpp | 12 const Tensor& elements, const Tensor& test_elements, bool invert, const Tensor& out) { in isin_default_kernel_gpu()
|
| /external/grpc-grpc/test/core/event_engine/posix/ |
| D | timer_heap_test.cc | 71 std::vector<Timer> test_elements = CreateTestElements(num_test_elements); in TEST() local
|
| /external/grpc-grpc/test/core/iomgr/ |
| D | timer_heap_test.cc | 77 grpc_timer* test_elements = create_test_elements(num_test_elements); in test1() local
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | TensorCompareKernel.cpp | 312 const Tensor& test_elements, in isin_default_kernel_cpu()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 2315 const std::vector<T>& test_elements) { in ExpectVectorEq()
2337 const std::set<T>& test_elements) { in ExpectSetEq()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | TensorCompare.cpp | 453 const Tensor& test_elements, in isin_sorting()
|
| /external/pytorch/torch/_decomp/ |
| D | decompositions.py | 5016 def isin(elements, test_elements, *, assume_unique=False, invert=False): argument
5031 def isin_default(elements, test_elements, *, invert=False): argument
5044 def isin_sorting(elements, test_elements, *, assume_unique=False, invert=False): argument
|
| /external/pytorch/torch/csrc/jit/runtime/static/ |
| D | generated_ops.cpp | 1000 const auto& test_elements = p_node->Input(1).toTensor(); in __anon7ebb59396402() local
1036 const auto& test_elements = p_node->Input(1).toTensor(); in __anon7ebb59396402() local
|
| /external/pytorch/torch/ |
| D | _meta_registrations.py | 6442 def meta_isin(elements, test_elements, *, assume_unique=False, invert=False): argument
|