Searched defs:tensor_indices_shapes (Results 1 – 2 of 2) sorted by relevance
120 … size_t *slice_nums, std::vector<ShapeVector> *tensor_indices_shapes) { in GetSliceShape()166 std::vector<ShapeVector> tensor_indices_shapes; in ConstGetTupleIndexInfo() local274 std::vector<ShapeVector> tensor_indices_shapes; in GetTupleIndexInfoInferInner() local
73 std::vector<ShapeVector> tensor_indices_shapes; in LaunchKernel() local