MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / ComputeGradSpans

Function ComputeGradSpans

tensorflow/core/kernels/scale_and_translate_op.cc:236–248  ·  view source on GitHub ↗

Computes the grad spans for the passed kernel. forward_input_size and forward_output_size are the input and output size from the forward operation.

Source from the content-addressed store, hash-verified

234// forward_input_size and forward_output_size are the input and output size from
235// the forward operation.
236Status ComputeGradSpans(OpKernelContext* context,
237 const functor::SamplingKernelType kernel_type,
238 const int64 forward_output_size,
239 const int64 forward_input_size, const float scale,
240 const float translate, const bool antialias,
241 Spans* grad_spans) {
242 Spans spans;
243 TF_RETURN_IF_ERROR(ComputeSpans(context, kernel_type, forward_output_size,
244 forward_input_size, scale, translate,
245 antialias, &spans));
246 return ComputeGradSpansCore(context, spans, forward_output_size,
247 forward_input_size, grad_spans);
248}
249
250void GetValues(OpKernelContext* context, int input_index, float* v_1,
251 float* v_2) {

Callers 1

ComputeMethod · 0.85

Calls 2

ComputeSpansFunction · 0.85
ComputeGradSpansCoreFunction · 0.85

Tested by

no test coverage detected