Searched refs:AsGraphDefForRewrite (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | compute_batch_size_op.cc | 101 OP_REQUIRES_OK(ctx, AsGraphDefForRewrite(ctx, dataset, &input_list, in Compute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | serialization_utils.h | 134 Status AsGraphDefForRewrite(OpKernelContext* ctx, const DatasetBase* input,
|
| H A D | rewrite_utils.cc | 178 AsGraphDefForRewrite(ctx, input, &input_list, &graph_def, &output_node)); in RewriteDataset()
|
| H A D | serialization_utils.cc | 420 Status AsGraphDefForRewrite(OpKernelContext* ctx, const DatasetBase* input, in AsGraphDefForRewrite() function
|