Searched refs:GetDatasetFromVariantTensor (Results 1 – 20 of 20) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset.cc | 145 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(tensor, &unused)); in Compute()
363 GetDatasetFromVariantTensor(tensor, &dataset).ok()) { in GetAllocatedBytes()
377 GetDatasetFromVariantTensor(tensor, &dataset).ok()) { in GetTotalBytes()
394 Status GetDatasetFromVariantTensor(const Tensor& tensor, in GetDatasetFromVariantTensor() function
508 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(t, &dataset)); in AddDatasetOrTensorHelper()
679 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &input)); in MakeDataset()
686 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &input)); in MakeDataset()
689 GetDatasetFromVariantTensor(ctx->input(1), &another_input)); in MakeDataset()
|
| D | dataset.h | 794 Status GetDatasetFromVariantTensor(const Tensor& tensor,
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_ops.cc | 83 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in Compute()
114 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in Compute()
|
| D | window_dataset_op_test.cc | 301 TF_ASSERT_OK(GetDatasetFromVariantTensor(window_dataset_tensor, in TEST_P()
495 TF_ASSERT_OK(GetDatasetFromVariantTensor(window_dataset_tensor, in TEST_P()
|
| D | rewrite_utils.cc | 180 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(outputs[0], rewritten_input)); in RewriteDataset()
|
| D | zip_dataset_op.cc | 226 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(i), &input)); in MakeDataset()
|
| D | iterator_ops.cc | 600 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in DoCompute()
633 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in DoCompute()
697 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in DoCompute()
909 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(return_values[0], &dataset)); in TryInit()
|
| D | captured_function.cc | 476 GetDatasetFromVariantTensor(return_values[0], &returned_dataset)); in MakeIteratorFromInputElement()
574 Status s = GetDatasetFromVariantTensor(t, &input); in AddToGraph()
|
| D | multi_device_iterator_ops.cc | 557 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in Compute()
|
| D | dataset_test_base.cc | 366 Status status = GetDatasetFromVariantTensor(*output, dataset); in GetDatasetFromContext()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | data_service_ops.cc | 42 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in Compute()
|
| D | directed_interleave_dataset_op.cc | 287 GetDatasetFromVariantTensor(ctx->input(0), &selector_input)); in MakeDataset()
302 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(i), &input)); in MakeDataset()
|
| D | to_tf_record_op.cc | 75 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in DoCompute()
|
| D | compute_batch_size_op.cc | 94 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in Compute()
|
| D | io_ops.cc | 46 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in DoCompute()
326 GetDatasetFromVariantTensor(reader_output[0], &input_)); in InitializeInput()
|
| D | choose_fastest_dataset_op.cc | 51 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(tensor, &input)); in MakeDataset()
|
| D | group_by_window_dataset_op.cc | 505 GetDatasetFromVariantTensor(return_values[0], &returned_dataset)); in StartFlushingGroup()
|
| D | snapshot_dataset_op.cc | 599 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(reader_output[0], &input_)); in Initialize()
|
| /external/tensorflow/tensorflow/core/data/ |
| D | standalone.cc | 87 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(outputs[0], &dataset)); in FromGraph()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_table_init_op.cc | 182 ctx, GetDatasetFromVariantTensor(ctx->input(1), &dataset), done); in ComputeAsync()
|