Home
last modified time | relevance | path

Searched refs:forward_input (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dwhile_v2_indexed_slices_rewriter.py69 for forward_input, output in zip(inputs_with_grads, structured_outputs):
72 if forward_input.dtype == dtypes.resource:
77 forward_input, loop_vars)
101 forward_input, loop_vars): argument
128 forward_input)
158 def _create_grad_indexed_slices_init(grad_output_slices, forward_input): argument
170 assert isinstance(forward_input, ops.Tensor)
181 if forward_input.dtype == dtypes.resource:
182 forward_shape = gen_resource_variable_ops.variable_shape(forward_input)
184 forward_shape = array_ops.shape(forward_input)
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dhierarchical_tree_broadcaster_test.cc343 void Broadcast(bool forward_input) { in Broadcast() argument
347 SchedClosure([di, forward_input, &done] { in Broadcast()
348 di->DoBroadcast(forward_input); in Broadcast()
409 bool forward_input) { in RunTest() argument
452 Broadcast(forward_input); in RunTest()
629 void DoBroadcast(bool forward_input) { in DoBroadcast() argument
653 if (forward_input) forward_from[0] = 0; in DoBroadcast()
Dexecutor.cc511 std::vector<int> forward_input; in InitializeNode() local
513 GetNodeAttr(n->attrs(), "_forward_input", &forward_input); in InitializeNode()
538 DCHECK_EQ(forward_input.size() % 2, 0); in InitializeNode()
539 for (int j = 0; j < forward_input.size(); j += 2) { in InitializeNode()
540 if (forward_input[j + 1] == i) { in InitializeNode()
542 forward_from[i] = forward_input[j]; in InitializeNode()
/external/tensorflow/tensorflow/core/kernels/
Dassign_op.h102 std::unique_ptr<Tensor> input_alias = context->forward_input( in Compute()
Dregex_replace_op.cc38 ctx->forward_input(0 /*input_index*/, 0 /*output_index*/, in InternalCompute()
Dlist_kernels.cc202 c->forward_input(input_index, output_index, DT_VARIANT, TensorShape{}, in ForwardInputOrCreateNewList()
430 std::unique_ptr<Tensor> maybe_result = c->forward_input( in Compute()
542 std::unique_ptr<Tensor> tl_alias = c->forward_input( in Compute()
Dunicode_ops.cc289 ctx->forward_input(0 /*input_index*/, 0 /*output_index*/, in Compute()
Dscatter_nd_op.cc199 std::unique_ptr<Tensor> forwarded_input = c->forward_input( in Compute()
Dresource_variable_ops.cc420 std::unique_ptr<Tensor> input_alias = context->forward_input( in Compute()
Dlist_kernels.h935 std::unique_ptr<Tensor> tls_alias = c->forward_input( in Compute()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc455 std::unique_ptr<Tensor> new_tensor = forward_input( in forward_input_to_output_with_shape()
496 std::unique_ptr<Tensor> OpKernelContext::forward_input( in forward_input() function in tensorflow::OpKernelContext
570 forward_input(input_index, Params::kNoReservation /*output_index*/, in forward_input_or_allocate_temp()
Dop_kernel.h850 std::unique_ptr<Tensor> forward_input(