Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_from_variant_op.cc31 const Tensor& encoded_variant, int ragged_rank, DataType value_dtype, in RaggedComponentsFromVariant() argument
33 const auto& flat_variants = encoded_variant.flat<Variant>(); in RaggedComponentsFromVariant()
208 const Tensor& encoded_variant = context->input(0); in Compute() local
212 input_ragged_rank_ = output_ragged_rank_ - encoded_variant.dims(); in Compute()
219 ", encoded_variant.dims(): ", encoded_variant.dims(), in Compute()
224 output_ragged_rank_ == encoded_variant.dims() + input_ragged_rank_, in Compute()
229 ", encoded_variant.dims(): ", encoded_variant.dims(), ".")); in Compute()
236 encoded_variant, input_ragged_rank_, in Compute()
240 if (encoded_variant.dims() == 0) { in Compute()
246 std::vector<int> encoded_dim_sizes(encoded_variant.dims(), 0); in Compute()
[all …]
Dragged_tensor_to_variant_op.cc188 Tensor encoded_variant = context->input(0); in Compute() local
196 const auto& flat_variants = encoded_variant.flat<Variant>(); in Compute()
Dragged_tensor_from_variant_op_test.cc83 auto encoded_variant = CreateVariantFromRagged<int, int64>( in TEST_F() local
97 {encoded_variant}); in TEST_F()
111 auto encoded_variant = CreateVariantFromRagged<int, int64>( in TEST_F() local
127 {encoded_variant}); in TEST_F()