Searched refs:tensor_length (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 160 const int tensor_length, const int num_ranks) = 0;
463 void InitExpected(std::vector<float>* expected, const int tensor_length, in InitExpected() argument
465 expected->resize(tensor_length); in InitExpected()
466 for (int i = 0; i < tensor_length; ++i) { in InitExpected()
498 void InitExpected(std::vector<float>* expected, const int tensor_length, in InitExpected() argument
500 expected->resize(tensor_length); in InitExpected()
501 for (int i = 0; i < tensor_length; ++i) { in InitExpected()
532 void InitExpected(std::vector<float>* expected, const int tensor_length, in InitExpected() argument
534 expected->resize(tensor_length * num_ranks, -1); in InitExpected()
536 for (int j = 0; j < tensor_length; ++j, ++i) { in InitExpected()
|
| D | ragged_tensor_to_tensor_op.cc | 106 const INDEX_TYPE tensor_length = row_split.size(); in GetMaxWidthRowSplit() local
107 if (tensor_length == 0 || tensor_length == 1) { in GetMaxWidthRowSplit()
111 for (INDEX_TYPE i = 0; i < tensor_length - 1; ++i) { in GetMaxWidthRowSplit()
|
| /external/pytorch/aten/src/ATen/test/ |
| D | mps_test_objc_interface.mm | 25 const unsigned int tensor_length = 100000UL;
30 torch::Tensor cpu_input1 = torch::randn({tensor_length}, at::device(at::kCPU));
31 torch::Tensor cpu_input2 = torch::randn({tensor_length}, at::device(at::kCPU));
36 torch::Tensor mps_output = torch::empty({tensor_length}, at::device(at::kMPS));
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| D | ragged_tensor_to_tensor_tflite.cc | 188 const int tensor_length = tensor_shape.FlatSize(); in GetMaxWidthRowSplit() local
189 if (tensor_length == 0 || tensor_length == 1) { in GetMaxWidthRowSplit()
205 for (int i = 1; i < tensor_length; ++i) { in GetMaxWidthRowSplit()
|
| /external/pytorch/test/ |
| D | test_cpp_extensions_aot.py | 99 tensor_length = 100000
100 x = torch.randn(tensor_length, device="cpu", dtype=torch.float32)
101 y = torch.randn(tensor_length, device="cpu", dtype=torch.float32)
|
| D | test_cpp_extensions_jit.py | 140 tensor_length = 100000
141 x = torch.randn(tensor_length, device="cpu", dtype=torch.float32)
142 y = torch.randn(tensor_length, device="cpu", dtype=torch.float32)
|