Searched refs:param_arrays (Results 1 – 2 of 2) sorted by relevance
2263 std::vector<IrArray> param_arrays; in ConstructIrArrayForInputs() local2264 param_arrays.reserve(hlo.operands().size()); in ConstructIrArrayForInputs()2266 param_arrays.push_back(GetIrArray(*param, hlo)); in ConstructIrArrayForInputs()2268 return param_arrays; in ConstructIrArrayForInputs()2272 const HloInstruction& hlo, const std::vector<IrArray>& param_arrays, in ConstructInputReducedShapeAndCastInputIrArrayToShape() argument2291 param_arrays[id].CastToShape((*param_reduced_shapes)[id], &b_)); in ConstructInputReducedShapeAndCastInputIrArrayToShape()3056 std::vector<IrArray> param_arrays = ConstructIrArrayForInputs(*unnested_hlo); in EmitKernel() local3057 int64 num_params = param_arrays.size(); in EmitKernel()3110 *unnested_hlo, param_arrays, param_shmem_buffers, reduced_dims, in EmitKernel()
304 const std::vector<llvm_ir::IrArray>& param_arrays,