Searched refs:out_reshape (Results 1 – 2 of 2) sorted by relevance
85 TensorShape out_reshape() const;186 helper.out_reshape(), &tmp_out, alloc_attr));
22 TensorShape ReductionHelper::out_reshape() const { in out_reshape() function in tensorflow::ReductionHelper