| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | dataset.cc | 162 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(tensor, &unused)); in Compute()
425 GetDatasetFromVariantTensor(tensor, &dataset).ok()) { in GetAllocatedBytes()
439 GetDatasetFromVariantTensor(tensor, &dataset).ok()) { in GetTotalBytes()
456 Status GetDatasetFromVariantTensor(const Tensor& tensor, in GetDatasetFromVariantTensor() function
819 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(t, &dataset)); in AddDatasetOrTensorHelper()
1034 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &input)); in MakeDataset()
1041 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &input)); in MakeDataset()
1044 GetDatasetFromVariantTensor(ctx->input(1), &another_input)); in MakeDataset()
|
| H A D | dataset.h | 883 Status GetDatasetFromVariantTensor(const Tensor& tensor,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | dataset_ops.cc | 84 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in Compute()
137 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in Compute()
|
| H A D | window_dataset_op_test.cc | 306 TF_ASSERT_OK(GetDatasetFromVariantTensor(window_dataset_tensor, in TEST_P()
500 TF_ASSERT_OK(GetDatasetFromVariantTensor(window_dataset_tensor, in TEST_P()
|
| H A D | get_options_op.cc | 30 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &input)); in Compute()
|
| H A D | options_dataset_op.cc | 115 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &input)); in MakeDataset()
|
| H A D | iterator_ops.cc | 602 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in DoCompute()
655 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in DoCompute()
840 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(return_values[0], &dataset)); in TryInit()
|
| H A D | reduce_dataset_op.cc | 54 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in DoCompute()
|
| H A D | zip_dataset_op.cc | 279 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(i), &input)); in MakeDataset()
|
| H A D | multi_device_iterator_ops.cc | 649 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in Compute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | directed_interleave_dataset_op.cc | 337 GetDatasetFromVariantTensor(ctx->input(0), &selector_input)); in MakeDataset()
353 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(i), &input)); in MakeDataset()
|
| H A D | random_access_ops.cc | 32 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in DoCompute()
|
| H A D | io_ops.cc | 79 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in DoCompute()
540 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in MakeDataset()
714 GetDatasetFromVariantTensor(reader_output[0], &input_)); in InitializeInput()
|
| H A D | to_tf_record_op.cc | 77 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in DoCompute()
|
| H A D | compute_batch_size_op.cc | 96 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in Compute()
|
| H A D | data_service_ops.cc | 90 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in Compute()
|
| H A D | lookup_ops.cc | 195 ctx, GetDatasetFromVariantTensor(ctx->input(1), &dataset), done); in ComputeAsync()
|
| H A D | choose_fastest_dataset_op.cc | 51 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(tensor, &input)); in MakeDataset()
|
| H A D | group_by_window_dataset_op.cc | 514 GetDatasetFromVariantTensor(return_values[0], &returned_dataset)); in StartFlushingGroup()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | standalone.cc | 107 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(outputs[0], &dataset)); in FromGraph()
|
| H A D | rewrite_utils.cc | 206 GetDatasetFromVariantTensor(outputs[0], &rewritten_dataset)); in RewriteDataset()
|
| H A D | serialization_utils.cc | 378 if (GetDatasetFromVariantTensor(val, &dataset).ok()) { in WriteTensorInternal()
|
| H A D | snapshot_utils.cc | 658 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(i), &input)); in MakeDataset()
|
| H A D | captured_function.cc | 410 GetDatasetFromVariantTensor(return_values[0], &returned_dataset)); in MakeIteratorFromInputElement()
|
| H A D | dataset_test_base.cc | 371 Status status = GetDatasetFromVariantTensor(*output, dataset); in GetDatasetFromContext()
|