Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dragged_tensor_to_sparse_kernel.cc44 std::vector<ConstFlatSplits> rt_nested_splits; in Compute() local
158 std::vector<ConstFlatSplits> rt_nested_splits, in ValidateInputs()
222 const std::vector<ConstFlatSplits>& rt_nested_splits) { in IsCompleted()
H A Dragged_range_op.cc75 auto rt_nested_splits = rt_nested_splits_out->flat<SPLITS_TYPE>(); in Compute() local
H A Dragged_tensor_to_sparse_kernel_test.cc40 const std::vector<std::vector<int64_t>>& rt_nested_splits, in BuildRaggedTensorToSparseGraph()
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
H A Dragged_range_tflite.cc97 SPLITS_TYPE* rt_nested_splits = in EvalT() local