Searched refs:nrows_source (Results 1 – 1 of 1) sorted by relevance
1941 def first_layer(cls, nrows_source, nrows_target): argument1943 gather_index = _first_layer_gather_index(nrows_source, nrows_target)2984 def _first_layer_gather_index(nrows_source, nrows_target): argument3002 do_broadcast = math_ops.equal(nrows_source,3003 constant_op.constant(1, nrows_source.dtype))3004 nrows_equal = math_ops.equal(nrows_source, nrows_target)