Home
last modified time | relevance | path

Searched refs:context_result (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dexample_proto_fast_parsing.cc2281 Allocator* allocator, Result* context_result) { in ParseContextDenseFeatures() argument
2302 context_result->dense_values[t] = Tensor(allocator, dtype, dense_shape); in ParseContextDenseFeatures()
2304 Tensor& out = context_result->dense_values[t]; in ParseContextDenseFeatures()
2345 Result* context_result) { in ParseContextSparseFeatures() argument
2356 context_result->sparse_indices[t] = in ParseContextSparseFeatures()
2358 context_result->sparse_values[t] = Tensor(allocator, dtype, values_shape); in ParseContextSparseFeatures()
2359 context_result->sparse_shapes[t] = in ParseContextSparseFeatures()
2361 Tensor& out_values = context_result->sparse_values[t]; in ParseContextSparseFeatures()
2363 int64* out_indices = context_result->sparse_indices[t].flat<int64>().data(); in ParseContextSparseFeatures()
2364 auto out_shape = context_result->sparse_shapes[t].vec<int64>(); in ParseContextSparseFeatures()
[all …]
Dexample_proto_fast_parsing.h156 thread::ThreadPool* thread_pool, example::Result* context_result,
/external/tensorflow/tensorflow/core/kernels/
Dexample_parsing_ops.cc473 example::Result context_result, feature_list_result; in Compute() local
479 &context_result, &feature_list_result, &dense_feature_lengths, in Compute()
482 OP_REQUIRES_OK(ctx, WriteOutput(context_result, feature_list_result, in Compute()
685 Status WriteOutput(const example::Result& context_result, in WriteOutput() argument
720 context_dense_values.set(d, context_result.dense_values[d]); in WriteOutput()
727 context_sparse_indices.set(d, context_result.sparse_indices[d]); in WriteOutput()
728 context_sparse_values.set(d, context_result.sparse_values[d]); in WriteOutput()
729 context_sparse_shapes.set(d, context_result.sparse_shapes[d]); in WriteOutput()
753 context_ragged_values.set(d, context_result.ragged_values[d]); in WriteOutput()
754 context_ragged_splits.set(d, context_result.ragged_splits[d]); in WriteOutput()