| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | parallel_filter_dataset_op.cc | 342 std::vector<Tensor> input_element; in CallFunction() local
343 result->status = input_impl_->GetNext(ctx.get(), &input_element, in CallFunction()
349 result->return_values = input_element; in CallFunction()
368 ctx.get(), std::move(input_element), &result->predicate_values, in CallFunction()
375 [this, ctx, result](std::vector<Tensor> input_element) { in CallFunction() argument
377 ctx.get(), std::move(input_element), in CallFunction()
380 std::move(input_element)); in CallFunction()
|
| D | parallel_map_dataset_op.cc | 466 std::vector<Tensor> input_element; in CallFunction() local
467 result->status = input_impl_->GetNext(ctx.get(), &input_element, in CallFunction()
484 ctx.get(), std::move(input_element), &result->return_values, in CallFunction()
491 [this, ctx, result](std::vector<Tensor> input_element) { in CallFunction() argument
493 ctx.get(), std::move(input_element), &result->return_values, in CallFunction()
496 std::move(input_element)); in CallFunction()
|
| D | shuffle_dataset_op.cc | 396 std::vector<Tensor> input_element; in FillBuffer() local
399 input_impl_->GetNext(ctx, &input_element, &end_of_input_sequence)); in FillBuffer()
401 AddToShuffleBuffer(ctx, std::move(input_element)); in FillBuffer()
|
| /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/data/ |
| D | captured_function.h | 47 const std::vector<Tensor>& input_element, int64_t thread_index,
56 const std::vector<Tensor>& input_element, int64_t thread_index,
|
| D | captured_function.cc | 382 const std::vector<Tensor>& input_element, int64_t thread_index, in MakeIteratorFromInputElement() argument
385 return MakeIteratorFromInputElement(ctx, parent, input_element, thread_index, in MakeIteratorFromInputElement()
392 const std::vector<Tensor>& input_element, int64_t thread_index, in MakeIteratorFromInputElement() argument
399 ctx, input_element, &return_values, node)); in MakeIteratorFromInputElement()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | parse_example_dataset_op.cc | 589 std::vector<Tensor> input_element; in CallFunction() local
590 result->status = input_impl_->GetNext(ctx.get(), &input_element, in CallFunction()
605 [this, ctx, result](std::vector<Tensor> input_element) { in CallFunction() argument
606 return ParseExample(ctx.get(), std::move(input_element), in CallFunction()
609 std::move(input_element)); in CallFunction()
|
| D | map_and_batch_dataset_op.cc | 411 std::vector<Tensor> input_element; in CallFunction() local
414 input_impl_->GetNext(ctx.get(), &input_element, &end_of_input); in CallFunction()
482 instantiated_captured_func_->RunAsync(ctx.get(), std::move(input_element), in CallFunction()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | image_ops_test.py | 811 input_element = max(dst_x - 1, 1) * max(dst_y - 1, 1)
812 input_data = [[input_element] * dst_x] * dst_y
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 2982 TF_ASSIGN_OR_RETURN(llvm::Value* const input_element, in EmitElementalReduce()
2984 reduction_operands.push_back(input_element); in EmitElementalReduce()
|