Home
last modified time | relevance | path

Searched refs:grad_source (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dtensor_array_grad.py103 grad_source = _GetGradSource(grad)
106 .grad(source=grad_source, flow=flow))
129 grad_source = _GetGradSource(flow)
138 .grad(source=grad_source, flow=flow))
166 grad_source = _GetGradSource(grad)
169 .grad(source=grad_source, flow=flow))
190 grad_source = _GetGradSource(flow)
199 .grad(source=grad_source, flow=flow))
227 grad_source = _GetGradSource(grad)
230 .grad(source=grad_source, flow=flow))
[all …]
/aosp_15_r20/external/libvpx/vp8/encoder/
H A Dpickinter.c61 int grad_source = 0; in check_dot_artifact_candidate() local
91 grad_source = macroblock_corner_grad(target_last, stride, 0, 0, 1, 1); in check_dot_artifact_candidate()
92 if (grad_last >= threshold1 && grad_source <= threshold2) { in check_dot_artifact_candidate()
98 grad_source = macroblock_corner_grad(target_last, stride, 0, shift, 1, -1); in check_dot_artifact_candidate()
99 if (grad_last >= threshold1 && grad_source <= threshold2) { in check_dot_artifact_candidate()
105 grad_source = macroblock_corner_grad(target_last, stride, shift, 0, -1, 1); in check_dot_artifact_candidate()
106 if (grad_last >= threshold1 && grad_source <= threshold2) { in check_dot_artifact_candidate()
112 grad_source = in check_dot_artifact_candidate()
114 if (grad_last >= threshold1 && grad_source <= threshold2) { in check_dot_artifact_candidate()
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Doptimizer.py264 grad_source = GradSource(
270 self.grad_to_source[p.grad] = grad_source
272 install_guard(grad_source.make_guard(GuardBuilder.CONSTANT_MATCH))
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dcase_op.cc181 for (const string& grad_source : update.tensor_array_gradients_accessed) { in Compile() local
183 << grad_source; in Compile()
186 grad_source, b, &gradient)); in Compile()
H A Dif_op.cc70 for (const string& grad_source : update.tensor_array_gradients_accessed) { in PopulateTensorArrayGradients() local
72 << grad_source; in PopulateTensorArrayGradients()
75 grad_source, b, &gradient)); in PopulateTensorArrayGradients()
H A Dwhile_op.cc401 for (const string& grad_source : update.tensor_array_gradients_accessed) { in Compile() local
403 << grad_source; in Compile()
406 grad_source, builder, &gradient)); in Compile()