Home
last modified time | relevance | path

Searched refs:take_grad (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dconditional_accumulator_test.py84 result = self.evaluate(q.take_grad(1))
105 result = accums[i].take_grad(1).eval()
118 takeg_t = q.take_grad(1)
155 takeg_t = q.take_grad(1)
204 extract_t = q.take_grad(2)
235 takeg_t = q.take_grad(1)
244 takeg_t = q.take_grad(constant_op.constant(1))
263 takeg_t = q.take_grad(1)
272 takeg_t = q.take_grad(constant_op.constant(1))
297 takeg_t = q.take_grad(-1)
[all …]
Dsparse_conditional_accumulator_test.py368 def take_grad(): function
372 threads = [self.checkedThread(target=take_grad) for _ in range(10)]
409 def take_grad(): function
413 takeg_thread = self.checkedThread(target=take_grad)
625 self.evaluate(q.take_grad(1))
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-conditional-accumulator.pbtxt35 name: "take_grad"
Dtensorflow.-sparse-conditional-accumulator.pbtxt39 name: "take_grad"
Dtensorflow.sparse.-sparse-conditional-accumulator.pbtxt39 name: "take_grad"
/external/tensorflow/tensorflow/python/training/
Dsync_replicas_optimizer.py290 aggregated_grad.append(grad_accum.take_grad(
/external/tensorflow/tensorflow/python/ops/
Ddata_flow_ops.py1382 def take_grad(self, num_required, name=None): member in ConditionalAccumulator
1519 def take_grad(self, num_required, name=None): member in SparseConditionalAccumulator