Searched refs:ConcatGrad (Results 1 – 2 of 2) sorted by relevance
216 Status ConcatGrad(const AttrSlice& attrs, FunctionDef* g) { in ConcatGrad() function224 REGISTER_OP_GRADIENT("Concat", ConcatGrad);
117 std::vector<Tensor> ConcatGrad(int dim, const Tensor& x0, const Tensor& x1, in ConcatGrad() function165 TEST(ArrayGradTest, ConcatGrad) { in TEST() argument174 auto dx = ConcatGrad(1, x0, x1, dy); in TEST()