Searched refs:permuted_shape (Results 1 – 3 of 3) sorted by relevance
857 Shape permuted_shape = ShapeUtil::PermuteDimensions(permutation, shape()); in Transpose() local872 CHECK(LayoutUtil::IsDenseArray(permuted_shape)); in Transpose()873 Layout* layout = permuted_shape.mutable_layout(); in Transpose()878 Literal new_literal(permuted_shape); in Transpose()
985 auto permuted_shape = Shape(scope, permuted); in ProdGrad() local1032 auto y = Reshape(scope, Mul(scope, left, right), permuted_shape); in ProdGrad()
305 permuted_shape = array_ops.shape(permuted)313 math_ops.conj(left) * math_ops.conj(right), permuted_shape)