Searched refs:reshapes (Results 1 – 13 of 13) sorted by relevance
383 std::vector<HloInstruction*> reshapes(fused_fusion_instrs.size()); in CreateFusedComputation() local388 reshapes[j], in CreateFusedComputation()395 TF_ASSIGN_OR_RETURN(auto concated_output, MakeConcatHlo(reshapes, 0)); in CreateFusedComputation()
405 std::vector<FDH::Node> reshapes = { in GradForBinaryCwise() local412 nodes.insert(nodes.end(), reshapes.begin(), reshapes.end()); in GradForBinaryCwise()
67 # shape `[]` reshapes to a scalar
55 fingerprint values are invariant under reshapes and bitcasts as long as the
132 This operation reshapes the "batch" dimension 0 into `M + 1` dimensions of shape
203 reshapes of each other, and also each output is a reshape of the
193 An example is a transformation that eliminates reshapes when they are redundant,
240 # Takes a scalar string and reshapes to a rank-1, single element string.
621 // Fold reshapes re-inserting reduced dimensions into the results of a reduction
3173 This operation reshapes the "batch" dimension 0 into space dimensions.3197 This operation reshapes space dimensions into the "batch" dimension 0
1091 This operation reshapes the "batch" dimension 0 into `M + 1` dimensions of shape10742 # shape `[]` reshapes to a scalar
3178 * Add meta-distribution BatchReshape which reshapes batch dimensions.