Searched defs:grad_index (Results 1 – 6 of 6) sorted by relevance
77 for (size_t grad_index = grad_start; grad_index < param_start; in execute_forward_backward() local
70 if (auto grad_index = in PropagateRequiresGradSimpleNode() local
552 if (auto grad_index = node->schema().argumentIndexWithName("dtype")) { in propagateTorchTensorShape() local
239 def grad_index(self): member in _GradLoopState
998 for (int grad_index = 0, end = grad.size(); grad_index < end; ++grad_index) { in ForwardpropFromTape() local
685 int grad_index = nir_tex_instr_src_index(tex, nir_tex_src_ddx); in lower_gles_arrayshadow_offset_filter() local