Searched refs:ComputeBitcastDimMap (Results 1 – 2 of 2) sorted by relevance
355 static std::optional<std::vector<std::vector<int64_t>>> ComputeBitcastDimMap(
1071 AlgebraicSimplifierVisitor::ComputeBitcastDimMap(const Shape& bitcast_shape, in ComputeBitcastDimMap() function in xla::AlgebraicSimplifierVisitor1266 auto dim_map = ComputeBitcastDimMap(bitcast->shape(), copy->shape()); in SwapCopyBitcastCopy()