Home
last modified time | relevance | path

Searched refs:input_element (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_map_dataset_op.cc426 std::vector<Tensor> input_element; in CallFunction() local
427 result->status = input_impl_->GetNext(ctx.get(), &input_element, in CallFunction()
444 ctx.get(), std::move(input_element), &result->return_values, in CallFunction()
451 [this, ctx, result](std::vector<Tensor> input_element) { in CallFunction() argument
453 ctx.get(), std::move(input_element), &result->return_values, in CallFunction()
456 std::move(input_element)); in CallFunction()
Dcaptured_function.h47 const std::vector<Tensor>& input_element, int64 thread_index,
56 const std::vector<Tensor>& input_element, int64 thread_index,
Dshuffle_dataset_op.cc180 std::vector<Tensor> input_element; in GetNextInternal() local
184 TF_RETURN_IF_ERROR(input_impl_->GetNext(ctx, &input_element, in GetNextInternal()
212 this->RecordBufferEnqueue(ctx, input_element); in GetNextInternal()
214 std::move(input_element); in GetNextInternal()
Dcaptured_function.cc448 const std::vector<Tensor>& input_element, int64 thread_index, in MakeIteratorFromInputElement() argument
451 return MakeIteratorFromInputElement(ctx, parent, input_element, thread_index, in MakeIteratorFromInputElement()
458 const std::vector<Tensor>& input_element, int64 thread_index, in MakeIteratorFromInputElement() argument
465 ctx, input_element, &return_values, node)); in MakeIteratorFromInputElement()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ReduceDataset.pbtxt20 A function that maps `(old_state, input_element)` to `new_state`. It must take
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dparse_example_dataset_op.cc583 std::vector<Tensor> input_element; in CallFunction() local
584 result->status = input_impl_->GetNext(ctx.get(), &input_element, in CallFunction()
599 [this, ctx, result](std::vector<Tensor> input_element) { in CallFunction() argument
600 return ParseExample(ctx.get(), std::move(input_element), in CallFunction()
603 std::move(input_element)); in CallFunction()
Dmap_and_batch_dataset_op.cc401 std::vector<Tensor> input_element; in CallFunction() local
404 input_impl_->GetNext(ctx.get(), &input_element, &end_of_input); in CallFunction()
472 instantiated_captured_func_->RunAsync(ctx.get(), std::move(input_element), in CallFunction()
/external/tensorflow/tensorflow/compiler/tests/
Dimage_ops_test.py702 input_element = max(dst_x - 1, 1) * max(dst_y - 1, 1)
703 input_data = [[input_element] * dst_x] * dst_y
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc2780 TF_ASSIGN_OR_RETURN(llvm::Value* const input_element, in EmitElementalReduce()
2782 reduction_operands.push_back(input_element); in EmitElementalReduce()