Searched refs:SplitVGrad (Results 1 – 2 of 2) sorted by relevance
247 Status SplitVGrad(const AttrSlice& attrs, FunctionDef* g) { in SplitVGrad() function266 REGISTER_OP_GRADIENT("SplitV", SplitVGrad);
241 std::vector<Tensor> SplitVGrad(const Tensor& x, const Tensor& size_splits, in SplitVGrad() function300 auto dx = SplitVGrad(x, size_splits, 1, dy0, dy1); in TEST()