Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_to_sparse_kernel.cc42 std::vector<ConstFlatSplits> rt_nested_splits; in Compute() local
155 std::vector<ConstFlatSplits> rt_nested_splits, in ValidateInputs()
211 const std::vector<ConstFlatSplits>& rt_nested_splits) { in IsCompleted()
Dragged_range_op.cc74 auto rt_nested_splits = rt_nested_splits_out->flat<SPLITS_TYPE>(); in Compute() local
Dragged_tensor_to_sparse_kernel_test.cc40 const std::vector<std::vector<int64>>& rt_nested_splits, in BuildRaggedTensorToSparseGraph()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_range_tflite.cc97 SPLITS_TYPE* rt_nested_splits = in EvalT() local