Home
last modified time | relevance | path

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

/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_range_tflite.cc92 TfLiteTensor& rt_nested_splits_out = in EvalT() local
95 context->ResizeTensor(context, &rt_nested_splits_out, in EvalT()
98 GetTensorData<SPLITS_TYPE>(&rt_nested_splits_out); in EvalT()
167 TfLiteTensor& rt_nested_splits_out = in Eval() local
169 switch (rt_nested_splits_out.type) { in Eval()
/external/tensorflow/tensorflow/core/kernels/
Dragged_range_op.cc70 Tensor* rt_nested_splits_out = nullptr; in Compute() local
73 &rt_nested_splits_out)); in Compute()
74 auto rt_nested_splits = rt_nested_splits_out->flat<SPLITS_TYPE>(); in Compute()