Searched refs:ToShape (Results 1 – 8 of 8) sorted by relevance
106 static TensorShape ToShape(const BCast::Vec& vec);
167 TensorShape BCast::ToShape(const BCast::Vec& vec) { in ToShape() function in tensorflow::BCast
65 const TensorShape output_shape = BCast::ToShape(bcast.output_shape()); in BinaryOpState()
57 shape_in = BCast::ToShape(bcast.output_shape()); in Compute()
136 TensorShape output_shape = BCast::ToShape(bcast.result_shape()); in Compute()
301 0, BCast::ToShape(bcast.output_shape()), &z)); in Compute()
474 0, BCast::ToShape(bcast.output_shape()), &z)); in Compute()
51 TensorShape bcast_shape = BCast::ToShape(bcast.output_shape()); in Compile()