Searched refs:inst_captured_func (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | take_while_dataset_op.cc | 58 InstantiatedCapturedFunction* inst_captured_func, in MakeDataset()
62 inst_captured_func->RunWithBorrowedArgs(ctx, args, &result)); in MakeDataset()
76 InstantiatedCapturedFunction* inst_captured_func, in MakeDataset() argument
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | filter_dataset_op.cc | 59 InstantiatedCapturedFunction* inst_captured_func, in MakeDataset()
63 inst_captured_func->RunWithBorrowedArgs(ctx, args, &result)); in MakeDataset()
77 InstantiatedCapturedFunction* inst_captured_func, in MakeDataset() argument
|
| D | map_dataset_op.cc | 58 InstantiatedCapturedFunction* inst_captured_func, in MakeDataset()
61 return inst_captured_func->Run(ctx, std::move(args), out_tensors); in MakeDataset()
68 InstantiatedCapturedFunction* inst_captured_func, in MakeDataset() argument
|
| D | dataset_utils.cc | 85 int64 thread_index, const InstantiatedCapturedFunction& inst_captured_func, in MakeIteratorFromInputElement() argument
89 TF_RETURN_IF_ERROR(inst_captured_func.RunWithBorrowedArgs(ctx, input_element, in MakeIteratorFromInputElement()
|
| D | dataset_utils.h | 47 int64 thread_index, const InstantiatedCapturedFunction& inst_captured_func,
|