Home
last modified time | relevance | path

Searched defs:params_nested_splits (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dragged_gather_op.cc135 std::vector<ConstFlatType> params_nested_splits; in MakeSplits() local
195 const std::vector<ConstFlatType>& params_nested_splits, in ValidateSplits()
H A Dragged_gather_op_test.cc36 const std::vector<std::vector<int64_t>>& params_nested_splits, in BuildRaggedGatherGraph()