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()
|