Home
last modified time | relevance | path

Searched refs:ragged_values_shape (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_to_variant_op_test.cc43 const TensorShape& ragged_values_shape, in BuildEncodeRaggedTensorGraph() argument
62 AddInputFromArray<VALUE_TYPE>(ragged_values_shape, ragged_values); in BuildEncodeRaggedTensorGraph()
68 const TensorShape& ragged_values_shape, in CreateVariantFromRagged() argument
78 Tensor values(DataTypeToEnum<VALUE_TYPE>::v(), ragged_values_shape); in CreateVariantFromRagged()
Dragged_tensor_from_variant_op_test.cc61 const TensorShape& ragged_values_shape, in CreateVariantFromRagged() argument
71 Tensor values(DataTypeToEnum<VALUE_TYPE>::v(), ragged_values_shape); in CreateVariantFromRagged()