Home
last modified time | relevance | path

Searched refs:GetDatasetFromVariantTensor (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Ddataset.cc122 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(tensor, &unused)); in Compute()
324 GetDatasetFromVariantTensor(tensor, &dataset).ok()) { in GetAllocatedBytes()
333 Status GetDatasetFromVariantTensor(const Tensor& tensor, in GetDatasetFromVariantTensor() function
416 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &input)); in MakeDataset()
423 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &input)); in MakeDataset()
426 GetDatasetFromVariantTensor(ctx->input(1), &another_input)); in MakeDataset()
Ddataset.h578 Status GetDatasetFromVariantTensor(const Tensor& tensor,
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_ops.cc32 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in Compute()
55 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in Compute()
Diterator_ops.cc164 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(outputs[0], &dataset)); in Restore()
554 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in Compute()
577 ctx, GetDatasetFromVariantTensor(ctx->input(0), &dataset), done); in ComputeAsync()
657 ctx, GetDatasetFromVariantTensor(ctx->input(0), &dataset), done); in ComputeAsync()
904 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(return_values[0], &dataset)); in TryInit()
Dzip_dataset_op.cc34 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(i), &input)); in MakeDataset()
Dgraph_rewrite_dataset.cc74 GetDatasetFromVariantTensor(outputs[0], &optimized_input_)); in Optimize()
Dflat_map_dataset_op.cc99 Status s = GetDatasetFromVariantTensor(t, &input); in AsGraphDefInternal()
Dfilter_dataset_op.cc141 Status s = GetDatasetFromVariantTensor(t, &input); in AsGraphDefInternal()
Dparallel_map_dataset_op.cc152 Status s = GetDatasetFromVariantTensor(t, &input); in AsGraphDefInternal()
Dmap_dataset_op.cc150 Status s = GetDatasetFromVariantTensor(t, &input); in AsGraphDefInternal()
Ddataset_utils.cc101 GetDatasetFromVariantTensor(return_values[0], &returned_dataset)); in MakeIteratorFromInputElement()
Ddataset_test_base.cc123 Status status = GetDatasetFromVariantTensor(*output, dataset); in GetDatasetFromContext()
Dinterleave_dataset_op.cc125 Status s = GetDatasetFromVariantTensor(t, &input); in AsGraphDefInternal()
Dmulti_device_iterator_ops.cc500 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in Compute()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddirected_interleave_dataset_op.cc35 GetDatasetFromVariantTensor(ctx->input(0), &selector_input)); in MakeDataset()
50 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(i), &input)); in MakeDataset()
Dto_tf_record_op.cc68 ctx, GetDatasetFromVariantTensor(ctx->input(0), &dataset), done); in ComputeAsync()
Dgroup_by_window_dataset_op.cc472 GetDatasetFromVariantTensor(return_values[0], &returned_dataset)); in StartFlushingGroup()
503 Status s = GetDatasetFromVariantTensor(t, &input); in OtherArgumentsNodeAndType()
Dtake_while_dataset_op.cc144 Status s = GetDatasetFromVariantTensor(t, &input); in AsGraphDefInternal()
Dscan_dataset_op.cc123 Status s = GetDatasetFromVariantTensor(t, &input); in AsGraphDefInternal()
Dchoose_fastest_dataset_op.cc50 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(tensor, &input)); in MakeDataset()
Dgroup_by_reducer_dataset_op.cc418 Status s = GetDatasetFromVariantTensor(t, &input); in OtherArgumentsNodeAndType()
Dchoose_fastest_branch_dataset_op.cc267 Status s = GetDatasetFromVariantTensor(t, &input); in AsGraphDefInternal()
Dmap_and_batch_dataset_op.cc221 Status s = GetDatasetFromVariantTensor(t, &input); in AsGraphDefInternal()
/external/tensorflow/tensorflow/core/common_runtime/data/
Dstandalone.cc70 TF_RETURN_IF_ERROR(GetDatasetFromVariantTensor(outputs[0], &dataset)); in FromGraph()
/external/tensorflow/tensorflow/contrib/bigtable/kernels/
Dbigtable_kernels.cc223 ctx, GetDatasetFromVariantTensor(ctx->input(1), &dataset), done); in ComputeAsync()

12