Searched refs:permuted_shape (Results 1 – 3 of 3) sorted by relevance
685 Shape permuted_shape = in Transpose() local700 CHECK(LayoutUtil::IsDenseArray(permuted_shape)); in Transpose()701 Layout* layout = permuted_shape.mutable_layout(); in Transpose()706 Literal new_literal(permuted_shape); in Transpose()
952 auto permuted_shape = Shape(scope, permuted); in ProdGrad() local999 auto y = Reshape(scope, Mul(scope, left, right), permuted_shape); in ProdGrad()
172 permuted_shape = array_ops.shape(permuted)180 permuted_shape)