Searched defs:tensor_it (Results 1 – 7 of 7) sorted by relevance
70 Iterator tensor_it(inout, collapsed); in run_op() local
99 Iterator tensor_it(&tensor, window); in fill_tensor() local
233 for (auto tensor_it = ordered_tensor_inputs.rbegin(); in handleBlockAndSubblocks() local
172 Iterator tensor_it(&tensor, tensor_window); in TEST_CASE() local
607 auto tensor_it = tensor_tape.find(id); in InitialGradients() local
671 auto tensor_it = tensor.begin(); in propagate_xla_data_direct() local
690 auto tensor_it = waiting_tensors_.find(batch_key); in Compute() local