Home
last modified time | relevance | path

Searched refs:grad_tensor (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dgradient_checker.cc188 TF_Tensor* grad_tensor; in CalcNumericalGrad() local
189 TF_RETURN_IF_ERROR(GetValue(diff_quotient.get(), &grad_tensor)); in CalcNumericalGrad()
191 memcpy(&grad_data[0], TF_TensorData(grad_tensor), in CalcNumericalGrad()
192 TF_TensorByteSize(grad_tensor)); in CalcNumericalGrad()
193 TF_DeleteTensor(grad_tensor); in CalcNumericalGrad()
Dtape.h999 Gradient* grad_tensor = grad[grad_index]; in ForwardpropFromTape() local
1000 if (grad_tensor != nullptr) { in ForwardpropFromTape()
1001 int64_t tensor_id = vspace_.TensorId(grad_tensor); in ForwardpropFromTape()
1005 tensor_id, vspace_.TapeTensorFromGradient(grad_tensor)); in ForwardpropFromTape()
/external/tensorflow/tensorflow/core/kernels/
Dconditional_accumulator.h119 const Tensor* grad_tensor; in GetAndValidateTensorInputForApplyGrad() local
120 OP_REQUIRES_OK_BOOLEAN(ctx, ctx->input("gradient", &grad_tensor)); in GetAndValidateTensorInputForApplyGrad()
121 *tensor = grad_tensor; in GetAndValidateTensorInputForApplyGrad()
/external/tensorflow/tensorflow/compiler/tests/
Dconcat_ops_test.py145 grad_tensor = constant_op.constant(
147 grad = gradients_impl.gradients([c], inp_tensors, [grad_tensor])
172 grad_tensor = constant_op.constant(
174 grad = gradients_impl.gradients([c], inp_tensors, [grad_tensor])
200 grad_tensor = constant_op.constant(
202 grad = gradients_impl.gradients([c], inp_tensors, [grad_tensor])
237 grad_tensor = constant_op.constant(
239 grad = gradients_impl.gradients([c], inp_tensors, [grad_tensor])
/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dconcat_op_test.py213 grad_tensor = constant_op.constant(
215 grad = gradients_impl.gradients([c], inp_tensors, [grad_tensor])
242 grad_tensor = constant_op.constant(
244 grad = gradients_impl.gradients([c], inp_tensors, [grad_tensor])
270 grad_tensor = constant_op.constant(
272 grad = gradients_impl.gradients([c], inp_tensors, [grad_tensor])
301 grad_tensor = constant_op.constant(grad_inp.flatten(), shape=output_shape)
302 grad = gradients_impl.gradients([c], inp_tensors, [grad_tensor])
322 grad_tensor = constant_op.constant(
325 grad = gradients_impl.gradients([c], [x, y], [grad_tensor])
[all …]
Dshape_ops_test.py571 grad_tensor = constant_op.constant(
573 grad = gradients_impl.gradients([tiled], [a], [grad_tensor])[0]
587 grad_tensor = constant_op.constant(
589 grad = gradients_impl.gradients([tiled], [a], [grad_tensor])[0]
607 grad_tensor = constant_op.constant(
609 grad = gradients_impl.gradients([tiled], [a], [grad_tensor])[0]
622 grad_tensor = constant_op.constant(
624 grad = gradients_impl.gradients([tiled], [a], [grad_tensor])[0]
Dslice_op_test.py331 grad_tensor = constant_op.constant(grads)
332 grad = gradients_impl.gradients(slice_t, [a], grad_tensor)[0]
356 grad_tensor = constant_op.constant(grads)
357 grad = tape.gradient(slice_t, [a], grad_tensor)[0]
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_avgpooling_op.cc191 const Tensor& grad_tensor = in Compute() local
229 : is_pool2d ? TFShapeToMklDnnDimsInNCHW(grad_tensor.shape(), in Compute()
231 : TFShapeToMklDnnDimsInNCDHW(grad_tensor.shape(), in Compute()
278 grad_dnn_data.SetUsrMem(diff_dst_md, &grad_tensor); in Compute()
285 static_cast<T*>(const_cast<T*>(grad_tensor.flat<T>().data())); in Compute()
Dmkl_maxpooling_op.cc245 const Tensor& grad_tensor = in Compute() local
281 : is_pool2d ? TFShapeToMklDnnDimsInNCHW(grad_tensor.shape(), in Compute()
283 : TFShapeToMklDnnDimsInNCDHW(grad_tensor.shape(), in Compute()
327 grad_dnn_data.SetUsrMem(diff_dst_md, &grad_tensor); in Compute()
334 static_cast<T*>(const_cast<T*>(grad_tensor.flat<T>().data())); in Compute()
/external/pytorch/torch/distributed/algorithms/ddp_comm_hooks/
Doptimizer_overlap_hooks.py147 for grad_tensor, model_param in zip(gradient_tensors, model_params):
154 grad_tensor,
/external/tensorflow/tensorflow/python/ops/
Dimage_grad_test_base.py171 grad_tensor = tape.gradient(resized_tensor, input_tensor)
172 self.assertEqual(in_shape, list(grad_tensor.get_shape()))
176 grad_values = self.evaluate(grad_tensor)
/external/tensorflow/tensorflow/python/eager/
Dforwardprop_test.py409 grad_tensor = constant_op.constant(array_ops.zeros([100]))
410 grad_tensor_weak = weakref.ref(grad_tensor)
411 with forwardprop.ForwardAccumulator(x, grad_tensor) as acc:
413 del grad_tensor
/external/pytorch/test/profiler/
Dtest_memory_profiler.py103 grad_tensor: torch.Tensor,
121 if node.tag == ctx and key_matches_tensor(p_grad_key, grad_tensor):
/external/pytorch/test/
Dtest_expanded_weights.py154 grad_tensor = torch.randn(4, requires_grad=True, device=device)
159 set_grad_sample_if_exists(grad_tensor, test_fn)
Dtest_nn.py5514 grad_tensor = torch.rand(grid.shape, device=device)
5517 grad_tensor_cl = grad_tensor.contiguous(memory_format=memory_format_cl)
5524 grid.backward(grad_tensor)