Searched refs:forward_input (Results 1 – 12 of 12) sorted by relevance
/external/tensorflow/tensorflow/python/ops/ |
D | while_v2_indexed_slices_rewriter.py | 69 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/ |
D | hierarchical_tree_broadcaster_test.cc | 343 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()
|
D | executor.cc | 511 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/ |
D | assign_op.h | 102 std::unique_ptr<Tensor> input_alias = context->forward_input( in Compute()
|
D | regex_replace_op.cc | 38 ctx->forward_input(0 /*input_index*/, 0 /*output_index*/, in InternalCompute()
|
D | list_kernels.cc | 202 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()
|
D | unicode_ops.cc | 289 ctx->forward_input(0 /*input_index*/, 0 /*output_index*/, in Compute()
|
D | scatter_nd_op.cc | 199 std::unique_ptr<Tensor> forwarded_input = c->forward_input( in Compute()
|
D | resource_variable_ops.cc | 420 std::unique_ptr<Tensor> input_alias = context->forward_input( in Compute()
|
D | list_kernels.h | 935 std::unique_ptr<Tensor> tls_alias = c->forward_input( in Compute()
|
/external/tensorflow/tensorflow/core/framework/ |
D | op_kernel.cc | 455 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()
|
D | op_kernel.h | 850 std::unique_ptr<Tensor> forward_input(
|