Searched defs:BroadcastArgs (Results 1 – 3 of 3) sorted by relevance
25 void BroadcastArgs(const RuntimeShape& input1_shape, const T* input1_data, in BroadcastArgs() function
2038 TEST_F(OpTest, BroadcastArgs) { in TEST_F() argument
4855 func BroadcastArgs(scope *Scope, s0 tf.Output, s1 tf.Output) (r0 tf.Output) { func