Home
last modified time | relevance | path

Searched refs:RunWithBorrowedArgs (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dgenerator_dataset_op.cc130 TF_RETURN_IF_ERROR(instantiated_init_func_->RunWithBorrowedArgs( in GetNextInternal()
140 Status s = instantiated_next_func_->RunWithBorrowedArgs( in GetNextInternal()
152 TF_RETURN_IF_ERROR(instantiated_finalize_func_->RunWithBorrowedArgs( in GetNextInternal()
Dcaptured_function.h240 Status RunWithBorrowedArgs(IteratorContext* ctx,
248 Status RunWithBorrowedArgs(IteratorContext* ctx,
Dcaptured_function.cc464 TF_RETURN_IF_ERROR(inst_captured_func.RunWithBorrowedArgs( in MakeIteratorFromInputElement()
890 Status InstantiatedCapturedFunction::RunWithBorrowedArgs( in RunWithBorrowedArgs() function in tensorflow::data::InstantiatedCapturedFunction
893 return RunWithBorrowedArgs(ctx, args, ret, /*node=*/nullptr); in RunWithBorrowedArgs()
896 Status InstantiatedCapturedFunction::RunWithBorrowedArgs( in RunWithBorrowedArgs() function in tensorflow::data::InstantiatedCapturedFunction
Dfilter_dataset_op.cc150 TF_RETURN_IF_ERROR(instantiated_captured_func_->RunWithBorrowedArgs( in GetNextInternal()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dgroup_by_reducer_dataset_op.cc231 TF_RETURN_IF_ERROR(instantiated_key_func_->RunWithBorrowedArgs( in GetNextInternal()
277 TF_RETURN_IF_ERROR(instantiated_finalize_func_->RunWithBorrowedArgs( in GetNextInternal()
Dtake_while_dataset_op.cc153 TF_RETURN_IF_ERROR(instantiated_captured_func_->RunWithBorrowedArgs( in GetNextInternal()
Dio_ops.cc150 TF_RETURN_IF_ERROR(function->RunWithBorrowedArgs( in GetShardIndex()
Dgroup_by_window_dataset_op.cc241 TF_RETURN_IF_ERROR(instantiated_key_func_->RunWithBorrowedArgs( in GetNextInternal()
Dsnapshot_dataset_op.cc687 TF_RETURN_IF_ERROR(instantiated_shard_func_->RunWithBorrowedArgs( in GetShardIndex()