Searched refs:optional_tensor (Results 1 – 5 of 5) sorted by relevance
187 const std::optional<at::Tensor>& optional_tensor = optional_tensorlist_args[i][idx]; in cpu_fallback() local188 at::_copy_from_and_resize(cpu_tensors[idx].value(), optional_tensor.value()); in cpu_fallback()
107 auto optional_tensor = std::optional<at::Tensor>(); in addBiasForConvIfNone() local108 module.setattr("bias", std::move(optional_tensor)); in addBiasForConvIfNone()
425 for (auto& optional_tensor : toListOptionalTensor()) {426 optional_tensor.~optional();
269 const auto optional_tensor = record_keeper.getClass("TFL_TensorOfOrNone"); in EmitOperandNumbers() local281 if (arg_def->getDef()->isSubClassOf(optional_tensor)) { in EmitOperandNumbers()
456 for (auto& optional_tensor : toListOptionalTensor()) {457 optional_tensor.~optional();