Searched refs:ReshapeToRank3 (Results 1 – 1 of 1) sorted by relevance
531 static Status ReshapeToRank3(const Tensor& input, int batch_size, in ReshapeToRank3() function561 TF_RETURN_IF_ERROR(ReshapeToRank3(inputs[0], bcast.x_batch_size(), &lhs)); in ContractOperands()563 TF_RETURN_IF_ERROR(ReshapeToRank3(inputs[1], bcast.y_batch_size(), &rhs)); in ContractOperands()581 ReshapeToRank3(*output, bcast.output_batch_size(), &output_reshaped)); in ContractOperands()