Searched refs:grad_components (Results 1 – 2 of 2) sorted by relevance
132 std::vector<std::vector<GradComponent>> grad_components(forward_input_size); in ComputeGradSpansCore() local141 grad_components[input_index].push_back( in ComputeGradSpansCore()147 for (std::vector<GradComponent>& gc : grad_components) { in ComputeGradSpansCore()170 if (!grad_components[input_index].empty()) { in ComputeGradSpansCore()171 const int start_span = grad_components[input_index].front().index; in ComputeGradSpansCore()173 for (const GradComponent& gc : grad_components[input_index]) { in ComputeGradSpansCore()
3972 unsigned grad_components) in lower_sampler_logical_send_gen4() argument4022 for (unsigned i = 0; i < grad_components; i++) in lower_sampler_logical_send_gen4()4025 msg_end = offset(msg_end, bld, MAX2(grad_components, 2)); in lower_sampler_logical_send_gen4()4027 for (unsigned i = 0; i < grad_components; i++) in lower_sampler_logical_send_gen4()4030 msg_end = offset(msg_end, bld, MAX2(grad_components, 2)); in lower_sampler_logical_send_gen4()4079 unsigned grad_components) in lower_sampler_logical_send_gen5() argument4124 for (unsigned i = 0; i < grad_components; i++) { in lower_sampler_logical_send_gen5()4187 unsigned grad_components) in lower_sampler_logical_send_gen7() argument4288 if (i < grad_components) { in lower_sampler_logical_send_gen7()4430 const unsigned grad_components = inst->src[TEX_LOGICAL_SRC_GRAD_COMPONENTS].ud; in lower_sampler_logical_send() local[all …]