Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/eager/
Dgradient_checker.cc187 TF_Tensor* grad_tensor; in CalcNumericalGrad() local
188 TF_RETURN_IF_ERROR(GetValue(diff_quotient.get(), &grad_tensor)); in CalcNumericalGrad()
190 memcpy(&grad_data[0], TF_TensorData(grad_tensor), in CalcNumericalGrad()
191 TF_TensorByteSize(grad_tensor)); in CalcNumericalGrad()
192 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 tensor_id = vspace_.TensorId(grad_tensor); in ForwardpropFromTape()
1005 tensor_id, vspace_.TapeTensorFromGradient(grad_tensor)); in ForwardpropFromTape()
/external/tensorflow/tensorflow/core/kernels/
Dconditional_accumulator.h122 const Tensor* grad_tensor; in GetAndValidateTensorInputForApplyGrad() local
123 OP_REQUIRES_OK_BOOLEAN(ctx, ctx->input("gradient", &grad_tensor)); in GetAndValidateTensorInputForApplyGrad()
124 *tensor = grad_tensor; in GetAndValidateTensorInputForApplyGrad()
/external/tensorflow/tensorflow/compiler/tests/
Dconcat_ops_test.py150 grad_tensor = constant_op.constant(
152 grad = gradients_impl.gradients([c], inp_tensors, [grad_tensor])
177 grad_tensor = constant_op.constant(
179 grad = gradients_impl.gradients([c], inp_tensors, [grad_tensor])
205 grad_tensor = constant_op.constant(
207 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])
/external/tensorflow/tensorflow/python/kernel_tests/
Dconcat_op_test.py216 grad_tensor = constant_op.constant(
218 grad = gradients_impl.gradients([c], inp_tensors, [grad_tensor])
245 grad_tensor = constant_op.constant(
247 grad = gradients_impl.gradients([c], inp_tensors, [grad_tensor])
273 grad_tensor = constant_op.constant(
275 grad = gradients_impl.gradients([c], inp_tensors, [grad_tensor])
304 grad_tensor = constant_op.constant(grad_inp.flatten(), shape=output_shape)
305 grad = gradients_impl.gradients([c], inp_tensors, [grad_tensor])
325 grad_tensor = constant_op.constant(
328 grad = gradients_impl.gradients([c], [x, y], [grad_tensor])
[all …]
Dshape_ops_test.py575 grad_tensor = constant_op.constant(
577 grad = gradients_impl.gradients([tiled], [a], [grad_tensor])[0]
591 grad_tensor = constant_op.constant(
593 grad = gradients_impl.gradients([tiled], [a], [grad_tensor])[0]
611 grad_tensor = constant_op.constant(
613 grad = gradients_impl.gradients([tiled], [a], [grad_tensor])[0]
626 grad_tensor = constant_op.constant(
628 grad = gradients_impl.gradients([tiled], [a], [grad_tensor])[0]
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_avgpooling_op.cc181 const Tensor& grad_tensor = in Compute() local
219 : is_pool2d ? TFShapeToMklDnnDimsInNCHW(grad_tensor.shape(), in Compute()
221 : TFShapeToMklDnnDimsInNCDHW(grad_tensor.shape(), in Compute()
267 grad_dnn_data.SetUsrMem(diff_dst_md, &grad_tensor); in Compute()
274 static_cast<T*>(const_cast<T*>(grad_tensor.flat<T>().data())); in Compute()
Dmkl_maxpooling_op.cc239 const Tensor& grad_tensor = in Compute() local
275 : is_pool2d ? TFShapeToMklDnnDimsInNCHW(grad_tensor.shape(), in Compute()
277 : TFShapeToMklDnnDimsInNCDHW(grad_tensor.shape(), in Compute()
320 grad_dnn_data.SetUsrMem(diff_dst_md, &grad_tensor); in Compute()
327 static_cast<T*>(const_cast<T*>(grad_tensor.flat<T>().data())); in Compute()
/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dslice_op_test.py325 grad_tensor = constant_op.constant(grads)
326 grad = gradients_impl.gradients(slice_t, [a], grad_tensor)[0]
350 grad_tensor = constant_op.constant(grads)
351 grad = tape.gradient(slice_t, [a], grad_tensor)[0]
/external/tensorflow/tensorflow/python/ops/
Dimage_grad_test_base.py173 grad_tensor = tape.gradient(resized_tensor, input_tensor)
174 self.assertEqual(in_shape, list(grad_tensor.get_shape()))
178 grad_values = self.evaluate(grad_tensor)
/external/tensorflow/tensorflow/python/eager/
Dforwardprop_test.py391 grad_tensor = constant_op.constant(array_ops.zeros([100]))
392 grad_tensor_weak = weakref.ref(grad_tensor)
393 with forwardprop.ForwardAccumulator(x, grad_tensor) as acc:
395 del grad_tensor