Searched defs:iota_shape (Results 1 – 9 of 9) sorted by relevance
48 Shape iota_shape = in TopK() local95 Shape iota_shape = in TopKWithPartitions() local
157 auto iota_shape = input_shape; in ArgMinMax() local
298 Shape iota_shape = x_shape; in EinsumDiagonalMask() local
44 auto iota_shape = input_shape.ValueOrDie(); in Compile() local
151 auto iota_shape = input_shape; in DataOutputFastPath() local
64 Shape iota_shape = sort->operand(0)->shape(); in ExpandInstruction() local
99 xla::Shape iota_shape; in OneHot() local
4028 const Shape iota_shape = in PreprocessHlos() local
7037 auto iota_shape = llvm::to_vector<4>(batch_dims); in QRBlock() local7120 auto iota_shape = llvm::to_vector<4>(batch_dims); in ComputeWYRepresentation() local