Searched refs:input_element (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | parallel_map_dataset_op.cc | 212 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()
|
| D | parallel_map_iterator.cc | 221 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()
|
| D | shuffle_dataset_op.cc | 107 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()
|
| D | dataset_utils.cc | 84 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()
|
| D | dataset_utils.h | 46 IteratorContext* ctx, const std::vector<Tensor>& input_element,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ReduceDataset.pbtxt | 20 A function that maps `(old_state, input_element)` to `new_state`. It must take
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | map_and_batch_dataset_op.cc | 415 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/ |
| D | ir_emitter.cc | 1861 TF_ASSIGN_OR_RETURN(llvm::Value* const input_element, in EmitElementalReduce()
1864 *reduce->to_apply(), {Load(accumulator_addr), input_element}, in EmitElementalReduce()
|