Searched defs:bitcast_shape (Results 1 – 2 of 2) sorted by relevance
331 const Shape& bitcast_shape, const Shape& op_shape, in ValidateTilingOfBitcast()1071 AlgebraicSimplifierVisitor::ComputeBitcastDimMap(const Shape& bitcast_shape, in ComputeBitcastDimMap()1229 const Shape& original_shape, const Shape& bitcast_shape, in InvertBitcastDimMap()
6143 auto bitcast_shape = ShapeUtil::MakeShape(F32, {kFeature, kOutput}); in TEST_P() local6185 auto bitcast_shape = ShapeUtil::MakeShape(F32, {kFeature, kOutput}); in TEST_P() local