Home
last modified time | relevance | path

Searched refs:CombineRaggedTensorToTensorShapes (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dragged_to_dense_util_test.cc29 TEST(CombineRaggedTensorToTensorShapes, UnknownShapeUnknownValue) { in TEST() argument
37 TF_ASSERT_OK(CombineRaggedTensorToTensorShapes( in TEST()
43 TEST(CombineRaggedTensorToTensorShapes, UnknownShape) { in TEST() argument
51 TF_ASSERT_OK(CombineRaggedTensorToTensorShapes( in TEST()
59 TEST(CombineRaggedTensorToTensorShapes, UnknownShapeDenseValue) { in TEST() argument
68 TF_ASSERT_OK(CombineRaggedTensorToTensorShapes( in TEST()
H A Dragged_to_dense_util.h48 Status CombineRaggedTensorToTensorShapes(int ragged_rank,
H A Dragged_to_dense_util.cc40 tensorflow::Status CombineRaggedTensorToTensorShapes( in CombineRaggedTensorToTensorShapes() function
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
H A Dragged_tensor_to_tensor_tflite.cc228 RuntimeShape CombineRaggedTensorToTensorShapes( in CombineRaggedTensorToTensorShapes() function
263 RuntimeShape result_shape = CombineRaggedTensorToTensorShapes( in CalculateOutputSize()
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dragged_conversion_ops.cc272 TF_RETURN_IF_ERROR(CombineRaggedTensorToTensorShapes( in RaggedTensorToTensorShapeFn()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dragged_tensor_to_tensor_op.cc162 TF_RETURN_IF_ERROR(CombineRaggedTensorToTensorShapes( in CalculateOutputSize()