Searched refs:params_input (Results 1 – 2 of 2) sorted by relevance
868 void RestoreParams(const OpInputList params_input, in RestoreParams() argument872 CHECK(params_input.size() == num_params) in RestoreParams()873 << "Number of params mismatch. Expected " << params_input.size() in RestoreParams()878 CHECK(size == params_input[i].NumElements()) in RestoreParams()880 << params_input[i].NumElements(); in RestoreParams()881 auto data_src_ptr = StreamExecutorUtil::AsDeviceMemory<T>(params_input[i]); in RestoreParams()
3447 const std::vector<CType> params_input = {CType(1), CType(2), CType(3), in TestConvertGather() local3487 {"params", test::AsTensor<CType>(params_input)}, in TestConvertGather()