Home
last modified time | relevance | path

Searched defs:rt_dense_values (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dragged_conversion_ops.cc167 ShapeHandle rt_dense_values = c->input(num_splits); in RaggedTensorToSparseShapeFn() local
194 shape_inference::ShapeHandle rt_dense_values = c->input(num_splits); in RaggedTensorToVariantShapeFn() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dragged_range_op.cc108 auto rt_dense_values = rt_dense_values_out->flat<T>(); in Compute() local
H A Dragged_tensor_to_sparse_kernel_test.cc42 const std::vector<T>& rt_dense_values) { in BuildRaggedTensorToSparseGraph()
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
H A Dragged_range_tflite.cc119 T* rt_dense_values = GetTensorData<T>(&rt_dense_values_out); in EvalT() local