Home
last modified time | relevance | path

Searched defs:tensor_it (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuFillKernel.cpp70 Iterator tensor_it(inout, collapsed); in run_op() local
/aosp_15_r20/external/ComputeLibrary/examples/
H A Dneon_permute.cpp99 Iterator tensor_it(&tensor, window); in fill_tensor() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dfrozen_concat_linear.cpp233 for (auto tensor_it = ordered_tensor_inputs.rbegin(); in handleBlockAndSubblocks() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/NEON/UNIT/
H A DTensorAllocator.cpp172 Iterator tensor_it(&tensor, tensor_window); in TEST_CASE() local
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dtape.h607 auto tensor_it = tensor_tape.find(id); in InitialGradients() local
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DFunctionalTensorWrapper.cpp671 auto tensor_it = tensor.begin(); in propagate_xla_data_direct() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dbatch_kernels.cc690 auto tensor_it = waiting_tensors_.find(batch_key); in Compute() local