Searched refs:RunWithBorrowedArgs (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | generator_dataset_op.cc | 103 instantiated_init_func_->RunWithBorrowedArgs(ctx, {}, &state_)); in GetNextInternal()
112 Status s = instantiated_next_func_->RunWithBorrowedArgs(ctx, state_, in GetNextInternal()
|
| D | captured_function.h | 63 Status RunWithBorrowedArgs(IteratorContext* ctx,
|
| D | filter_dataset_op.cc | 63 inst_captured_func->RunWithBorrowedArgs(ctx, args, &result)); in MakeDataset()
|
| D | dataset_utils.cc | 89 TF_RETURN_IF_ERROR(inst_captured_func.RunWithBorrowedArgs(ctx, input_element, in MakeIteratorFromInputElement()
|
| D | captured_function.cc | 336 Status InstantiatedCapturedFunction::RunWithBorrowedArgs( in RunWithBorrowedArgs() function in tensorflow::data::InstantiatedCapturedFunction
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | group_by_reducer_dataset_op.cc | 219 TF_RETURN_IF_ERROR(instantiated_key_func_->RunWithBorrowedArgs( in GetNextInternal()
264 TF_RETURN_IF_ERROR(instantiated_finalize_func_->RunWithBorrowedArgs( in GetNextInternal()
|
| D | take_while_dataset_op.cc | 62 inst_captured_func->RunWithBorrowedArgs(ctx, args, &result)); in MakeDataset()
|
| D | group_by_window_dataset_op.cc | 221 TF_RETURN_IF_ERROR(instantiated_key_func_->RunWithBorrowedArgs( in GetNextInternal()
|