Searched refs:BroadcastToGrad (Results 1 – 2 of 2) sorted by relevance
553 Status BroadcastToGrad(const AttrSlice& attrs, FunctionDef* g) { in BroadcastToGrad() function577 REGISTER_OP_GRADIENT("BroadcastTo", BroadcastToGrad);
768 std::vector<Tensor> BroadcastToGrad(const Tensor& x, const Tensor& shape, in BroadcastToGrad() function792 TEST(ArrayGradTest, BroadcastToGrad) { in TEST() argument799 auto dx = BroadcastToGrad(x, shape, dy); in TEST()