Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_map_dataset_op.cc212 std::vector<Tensor> input_element, in MapFunc() argument
216 size_t num_args = input_element.size(); in MapFunc()
221 input_element[dataset_->short_circuit_indices_[i]])); in MapFunc()
224 input_element[dataset_->short_circuit_indices_[i]]); in MapFunc()
249 std::vector<Tensor> input_element, in MapFunc() argument
252 std::vector<Tensor> input_element, in MapFunc()
256 ctx, std::move(input_element), result, std::move(done), prefix); in MapFunc()
260 std::move(input_element), result, in MapFunc()
263 map_func(ctx, prefix, std::move(input_element), result, in MapFunc()
Dparallel_map_iterator.cc221 std::vector<Tensor> input_element; in CallFunction() local
223 input_impl_->GetNext(ctx.get(), &input_element, &result->end_of_input); in CallFunction()
237 std::move(input_element), in CallFunction()
Dshuffle_dataset_op.cc107 std::vector<Tensor> input_element; in GetNextInternal() local
111 TF_RETURN_IF_ERROR(input_impl_->GetNext(ctx, &input_element, in GetNextInternal()
132 this->RecordBufferEnqueue(ctx, input_element); in GetNextInternal()
134 std::move(input_element); in GetNextInternal()
Ddataset_utils.cc84 IteratorContext* ctx, const std::vector<Tensor>& input_element, in MakeIteratorFromInputElement() argument
89 TF_RETURN_IF_ERROR(inst_captured_func.RunWithBorrowedArgs(ctx, input_element, in MakeIteratorFromInputElement()
Ddataset_utils.h46 IteratorContext* ctx, const std::vector<Tensor>& input_element,
/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/
Dmap_and_batch_dataset_op.cc415 std::vector<Tensor> input_element; in CallFunction() local
418 input_impl_->GetNext(ctx.get(), &input_element, &end_of_input); in CallFunction()
488 std::move(input_element), std::move(return_values), in CallFunction()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc1861 TF_ASSIGN_OR_RETURN(llvm::Value* const input_element, in EmitElementalReduce()
1864 *reduce->to_apply(), {Load(accumulator_addr), input_element}, in EmitElementalReduce()