Searched refs:CastGrad (Results 1 – 3 of 3) sorted by relevance
378 Status CastGrad(const AttrSlice& attrs, FunctionDef* g) { in CastGrad() function392 REGISTER_OP_GRADIENT("Cast", CastGrad);
1142 Status CastGrad(const Scope& scope, const Operation& op, in CastGrad() function1161 REGISTER_GRADIENT_OP("Cast", CastGrad);
1005 TEST_F(NaryGradTest, CastGrad) { in TEST_F() argument