Searched refs:builder_idex (Results 1 – 1 of 1) sorted by relevance
301 size_t builder_idex, const std::vector<int64_t> &dyn_input_sizes, in SetInputKernelBuilderInfo() argument331 auto type_id = DtypeToTypeId(dtypes[builder_idex]); in SetInputKernelBuilderInfo()333 inputs_format.push_back(formats[builder_idex]); in SetInputKernelBuilderInfo()338 auto type_id = DtypeToTypeId(dtypes[builder_idex]); in SetInputKernelBuilderInfo()340 inputs_format.push_back(formats[builder_idex]); in SetInputKernelBuilderInfo()345 auto type_id = DtypeToTypeId(dtypes[builder_idex]); in SetInputKernelBuilderInfo()347 inputs_format.push_back(formats[builder_idex]); in SetInputKernelBuilderInfo()357 …OutputKernelBuilderInfo(const std::vector<std::shared_ptr<OpIOInfo>> &outputs, size_t builder_idex, in SetOutputKernelBuilderInfo() argument397 auto type_id = DtypeToTypeId(dtypes[builder_idex]); in SetOutputKernelBuilderInfo()399 outputs_format.push_back(formats[builder_idex]); in SetOutputKernelBuilderInfo()