MCPcopy
hub / github.com/tensorflow/tfjs / grad

Method grad

tfjs-core/src/engine.ts:509–520  ·  view source on GitHub ↗
(dy: Tensor)

Source from the content-addressed store, hash-verified

507 {x} as unknown as NamedTensorMap);
508 const inputs = {x};
509 const grad = (dy: Tensor) => ({
510 x: () => {
511 const dtype = 'float32';
512 const gradInputs = {x: dy};
513 const attrs = {dtype};
514
515 return ENGINE.runKernel(
516 Cast, gradInputs as unknown as NamedTensorMap,
517 // tslint:disable-next-line: no-unnecessary-type-assertion
518 attrs as unknown as NamedAttrMap) as Tensor;
519 }
520 });
521 const saved: Tensor[] = [];
522 this.addTapeNode(this.state.activeScope.name, inputs, [y], grad, saved, {});
523 return y;

Callers 15

gradients_test.tsFile · 0.80
fFunction · 0.80
tensor_test.tsFile · 0.80
round_test.tsFile · 0.80
acosh_test.tsFile · 0.80
zeros_like_test.tsFile · 0.80
reverse_2d_test.tsFile · 0.80
ones_like_test.tsFile · 0.80
avg_pool_test.tsFile · 0.80

Calls 1

runKernelMethod · 0.80

Tested by

no test coverage detected