Searched refs:full_result_shape (Results 1 – 4 of 4) sorted by relevance
196 StatusOr<XlaOp> RngBitGeneratorInternal(const Shape& full_result_shape,
354 const Shape& full_result_shape, RandomAlgorithm algorithm, in RngBitGeneratorInternal() argument357 full_result_shape, builder_)); in RngBitGeneratorInternal()
2234 const Shape& full_result_shape, RandomAlgorithm algorithm, in RngBitGeneratorInternal() argument2237 *instr.mutable_shape() = full_result_shape.ToProto(); in RngBitGeneratorInternal()
822 const Shape& full_result_shape, RandomAlgorithm algorithm,