Searched defs:matrix_index (Results 1 – 4 of 4) sorted by relevance
2109 for (size_t matrix_index = 0; matrix_index != num_phi_placeholders; ++matrix_index) { in MaterializeLoopPhis() local2145 for (size_t matrix_index = 0; matrix_index != num_phi_placeholders; ++matrix_index) { in MaterializeLoopPhis() local2161 for (uint32_t matrix_index : current_dependencies->Indexes()) { in MaterializeLoopPhis() local2168 for (uint32_t matrix_index = 0; matrix_index != num_phi_placeholders; ++matrix_index) { in MaterializeLoopPhis() local2176 for (uint32_t matrix_index = 0; matrix_index != num_phi_placeholders; ++matrix_index) { in MaterializeLoopPhis() local
128 void ComputeTensorSlice(OpKernelContext* context, int64_t matrix_index, in ComputeTensorSlice()
233 OpKernelContext* context, int64_t matrix_index, const TensorInputs& inputs, in ComputeTensorSlice()
201 size_t matrix_index = 0; in run() local