Home
last modified time | relevance | path

Searched refs:forwarded_input (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.cc309 int forwarded_input; in Compute() local
312 {0}, 0, input.shape(), &old_lhs, &forwarded_input)); in Compute()
313 if (forwarded_input < 0) { in Compute()
Dscatter_nd_op.cc204 std::unique_ptr<Tensor> forwarded_input = in Compute() local
207 if (forwarded_input == nullptr) { in Compute()
221 c, indices, updates, shape, forwarded_input.get(), in Compute()
224 c->set_output(0, *forwarded_input); in Compute()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h905 int* forwarded_input = nullptr) TF_MUST_USE_RESULT;
1638 const TensorShape& output_shape, Tensor** output, int* forwarded_input) { in forward_input_or_allocate_output() argument
1642 if (forwarded_input != nullptr) { in forward_input_or_allocate_output()
1643 *forwarded_input = input_index; in forward_input_or_allocate_output()
1648 if (forwarded_input != nullptr) { in forward_input_or_allocate_output()
1649 *forwarded_input = -1; in forward_input_or_allocate_output()
/external/tensorflow/tensorflow/c/
Dkernels.h352 int output_num_dims, int* forwarded_input, TF_Status* status);
Dkernels.cc486 int output_num_dims, int* forwarded_input, TF_Status* status) { in TF_ForwardInputOrAllocateOutput() argument
500 forwarded_input); in TF_ForwardInputOrAllocateOutput()
Dkernels_test.cc930 int forwarded_input; in TEST_F() local
936 &forwarded_input, /*status=*/s); in TEST_F()
938 EXPECT_EQ(forwarded_input, 0); in TEST_F()